训练出错，ImportError: cannot import name '_DataLoaderIter'

[zyf1040895256]

感谢您的工作，我有一个问题。当我用DF2K训练时，发现报错信息：

Traceback (most recent call last): File "main.py", line 4, in import data File "/home/XXXX/F/XXX/DASR/data/init.py", line 2, in from dataloader import MSDataLoader File "/home/XXXX/F/XXX/DASR/dataloader.py", line 12, in from torch.utils.data.dataloader import _DataLoaderIter ImportError: cannot import name '_DataLoaderIter'

我的环境是：

torch 1.1.0
能解决下吗？

[LongguangWang]

Hi @zyf1040895256, 感谢对我们工作的关注。 首先，请再次确认下运行DASR代码时使用的环境是不是pytorch 1.1.0。在1.1.0版本应该不会出现_DataLoaderIter载入错误的。 其次，可以尝试将_DataLoaderIter类(连接)复制到你的torch.utils.data.dataloader.py中，应该也可以解决这个问题。 另外，也可以参考这个回答将DASR中数据读取部分的代码替换为EDSR 1.2.0版本中的数据读取代码。