RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1)

[zwr9cu]

excuse me, I get this error when I train coco with pytorch==0.4 .1 python==3.5.2 , I change batch_size from 20 to 10.how can i solve the problem? thanks

File "/home/ubuntu/MyFiles/zwr/NeuralBabyTalk/misc/utils.py", line 496, in bbox_overlaps
max_bbox = gt_box.size(1)
RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1)
Exception ignored in: <bound method _DataLoaderIter.del of <torch.utils.data.dataloader._DataLoaderIter object at 0x7fe36cfb1198>>