当模型多GPU运行时,遇到dimension specified as 0 but tensor has no dimensions问题
当使用pytorch多gpu运行时候,loss反传时出现错误:
Traceback (most recent call last):
File "trainval_net.py", line 343, in <module>
rois_label = fasterRCNN(im1_data,im2_data, im1_info, im2_info,gt_boxes, num...
原创
2019-04-10 09:26:11 ·
5219 阅读 ·
2 评论