search

NVlabs / SegFormer

Official PyTorch implementation of SegFormer
https://arxiv.org/abs/2105.15203
Other
2.58k stars 357 forks source link

raise StopIteration #147

Open dawang-11 opened 8 months ago

dawang-11 commented 8 months ago

Traceback (most recent call last): File "D:\anaconda3\envs\mm\lib\site-packages\mmcv\runner\iter_based_runner.py", line 32, in next data = next(self.iter_loader) File "D:\anaconda3\envs\mm\lib\site-packages\torch\utils\data\dataloader.py", line 517, in next data = self._next_data() File "D:\anaconda3\envs\mm\lib\site-packages\torch\utils\data\dataloader.py", line 1172, in _next_data raise StopIteration StopIteration

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "tools/train.py", line 169, in main() File "tools/train.py", line 158, in main train_segmentor( File "d:\segformer-master\mmseg\apis\train.py", line 115, in train_segmentor runner.run(data_loaders, cfg.workflow) File "D:\anaconda3\envs\mm\lib\site-packages\mmcv\runner\iter_based_runner.py", line 131, in run iter_runner(iter_loaders[i], **kwargs) File "D:\anaconda3\envs\mm\lib\site-packages\mmcv\runner\iter_based_runner.py", line 58, in train data_batch = next(data_loader) File "D:\anaconda3\envs\mm\lib\site-packages\mmcv\runner\iter_based_runner.py", line 38, in next data = next(self.iter_loader) File "D:\anaconda3\envs\mm\lib\site-packages\torch\utils\data\dataloader.py", line 517, in next data = self._next_data() File "D:\anaconda3\envs\mm\lib\site-packages\torch\utils\data\dataloader.py", line 1172, in _next_data raise StopIteration StopIteration