raise StopIteration StopIteration

[Aruen24]

Traceback (most recent call last): File "train.py", line 178, in train() File "train.py", line 134, in train images, targets = next(batch_iterator) File "/home/wyw/.conda/envs/py36_torch/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 435, in next data = self._next_data() File "/home/wyw/.conda/envs/py36_torch/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 1058, in _next_data raise StopIteration StopIteration

batch_iterator = iter(data.DataLoader(dataset, batch_size, shuffle=True, num_workers=num_workers,
collate_fn=detection_collate, drop_last=True))
change to
batch_iterator = iter(data.DataLoader(dataset, batch_size, shuffle=True, num_workers=num_workers,
collate_fn=detection_collate))