RuntimeError: Expected object of scalar type Long but got scalar type Float for argument #2 'ta

在用pytorch进行深度神经网络训练时,出现以下问题,特做记录

RuntimeError: Expected object of scalar type Long but got scalar type Float for argument #2 'ta

这个错误是说需要的数据类型是long而我的数据类型是float,所以需要进行数据类型转化

在pytorch中数据转化有专属自己的用法,即data=data.long()

所以在代码中做了如下修改:

cond_op_truth_var = cond_op_truth_var.long()

问题解决

 

 

 

上一篇:在图像中绘制基本形状和文字


下一篇:Scalar元组