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

The-Learning-And-Vision-Atelier-LAVA / DASR

[CVPR 2021] Unsupervised Degradation Representation Learning for Blind Super-Resolution

MIT License

395 stars 50 forks source link

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

Open zyf1040895256 opened 3 years ago

zyf1040895256 commented 3 years ago

感谢您的工作，我有一个问题。当我用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 commented 3 years ago

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