ImportError: cannot import name ‘_DataLoaderIter‘ from ‘torch.utils.data.dataloader‘

JOYCE_Leo16

发布于 2024-04-02 09:18:50

2600

发布于 2024-04-02 09:18:50

文章被收录于专栏：计算机视觉

复现代码过程中遇到报错：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可以不用修改。

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2024-04-02，如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度