IndexError happend in training

[FanShuixing]

Hi,Thanks for your sharing.I met a problem as follows when i run train.py
train.py:125: UserWarning: volatile was removed and now has no effect. Use with torch.no_grad():instead. img = Variable(img, volatile=True).cuda() train.py:126: UserWarning: volatile was removed and now has no effect. Usewith torch.no_grad(): instead. gt_map = Variable(gt_map, volatile=True).cuda() Traceback (most recent call last): File "train.py", line 205, in <module> main() File "train.py", line 81, in main validate(val_loader, net, epoch, restore_transform) File "train.py", line 141, in validate gt_count_tmp = gt_count[i_img] IndexError: index 1 is out of bounds for axis 0 with size 1

This happend in validate(),is there something wrong in this function? I have print the gt_count.shape,it equals to (1,),and pred_map.shape equals to (768,1024),and i don't understand why use gt_count_tmp=gt_count[0] in line 141 of train.py.
Thanks for your reply.

[2king2]

Hello, I also ran into this problem, I ask you to solve it? How'd it work out