问题描述
复现代码过程中遇到报错:ImportError: cannot import name '_DataLoaderIter' from 'torch.utils.data.dataloader' 。其中这个问题之前也遇到过,但是忘记是哪个模型了。
解决方案
将下面代码:
from torch.utils.data.dataloader import _DataLoaderIter
修改为:
from torch.utils.data.dataloader import _SingleProcessDataLoaderIter from torch.utils.data.dataloader import _MultiProcessingDataLoaderIter
这是由于torch版本问题引发的错误,pytorch环境是torch1.1.0可以不用修改。