hello, I encounter a problem when training, please help me
Closed this issue · 1 comments
zhengkai17 commented
when I train the model on VOC2012, it stops at the 160/20000 .
Traceback (most recent call last):
File "train.py", line 474, in
main()
File "train.py", line 335, in main
_, batch = trainloader_remain_iter.next()
File "/home/zhengkai/anaconda3/envs/tensorflow/lib/python3.5/site-packages/torch/utils/data/dataloader.py", line 572, in next
raise StopIteration
StopIteration
hfslyc commented
Hi, it seems that your dataloader has problem during data loading. You might want to check what's the error inside dataloader (like missing file?)