I am using pytorch 0.3.0.post4. I try to train a ssd model on coco using codes from this repo 32 and I get this stopiteration stuff and following messages after around 1000 iterations. I was wondering what would cause this problem and is there any idea to solve it?
File “train.py”, line 255, in 
train()
File “train.py”, line 165, in train
images, targets = next(batch_iterator)
File “/home/rusu5516/.local/lib/python3.5/site-packages/torch/utils/data/dataloader.py”, line 200, in next
raise StopIteration
StopIteration