- 博客(2)
- 收藏
- 关注
原创 RuntimeError: Expected object of scalar type Long but got scalar type Float for argument #2 'target'
在用pytorch进行深度神经网络训练时,出现以下问题,特做记录这个错误是说需要的数据类型是long而我的数据类型是float,所以需要进行数据类型转化在pytorch中数据转化有专属自己的用法,即data=data.long()所以在代码中做了如下修改:cond_op_truth_var = cond_op_truth_var.long()问题解决...
2019-06-04 10:26:54
8807
3
原创 IndexError:too many indices for array
在用pytorch进行深度学习训练时,出现如下错误:经过查询,发现该错误是由于pytorch升级引起的,我使用的版本为pytorch 0.40版本,所以做了如下修改,问题解决if torch.__version__.startswith('0.3'): bbox_loss += net.bbox_loss.data.cpu().numpy()[0] iou_los...
2019-06-04 10:20:10
1836
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人