- 博客(2)
- 收藏
- 关注
RSS订阅原创 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
8787
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
1829
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人