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

[CVPR 2021] Exploring Sparsity in Image Super-Resolution for Efficient Inference
238 stars 30 forks source link

ImportError: cannot import name '_DataLoaderIter #13

Open wusuhuang opened 3 years ago

wusuhuang commented 3 years ago

ImportError: cannot import name '_DataLoaderIter' from 'torch.utils.data.dataloader' (/opt/conda/lib/python3.8/site-packages/torch/utils/data/dataloader.py)

your pytorch version is 1.1.0, i should change my pytorch version to 1.1.0?

LongguangWang commented 3 years ago

Hi @wusuhuang, it is recommanded to change the pytorch version to 1.1.0. Or you can try the solution here, maybe it can also solve this issue.