langmanbusi / Semantic-Aware-Low-Light-Image-Enhancement

Semantic-Aware LLIE. CVPR 2023
MIT License
194 stars 7 forks source link

环境配置 #29

Closed Xu1103 closed 3 months ago

Xu1103 commented 5 months ago

配置好pytorch为1.8.0后,运行DRBN_SKF的测试数据时出错,Traceback (most recent call last): File "main_test.py", line 4, in import data File "/Projects_xkh/Semantic-Aware-Low-Light-Image-Enhancement-main/DRBN_SKF/src/data/init.py", line 3, in from dataloader import MSDataLoader File "/Projects_xkh/Semantic-Aware-Low-Light-Image-Enhancement-main/DRBN_SKF/src/dataloader.py", line 13, in from torch.utils.data.dataloader import _DataLoaderIter ImportError: cannot import name '_DataLoaderIter' from 'torch.utils.data.dataloader',请问这个环境版本过高的问题该怎么解决啊?

langmanbusi commented 5 months ago

可以把这个类复制进去,我就是这么干的 https://github.com/pytorch/pytorch/blob/0b868b19063645afed59d6d49aff1e43d1665b88/torch/utils/data/dataloader.py#L199