使用torch.utils.data.DataLoader()时出现 “Cannot find reference ‘data’in __init__.py“ 问题

使用时情况如下:

使用torch.utils.data.DataLoader()时出现 “Cannot find reference ‘data’in __init__.py“ 问题_第1张图片

我们直接使用torch.utils.data.DataLoader()时,DataLoader函数在使用时PyCharm黄色高亮并显示报错,同时,我输入DataLoader时,也没有提示语句,很奇怪。并且,对于我的pytorch 1.10.1版本来说,上述代码运行是可以运行的,也没有报错,只是PyCharm这个IDLE出现问题。

解决办法:将torch.utils.data在初始时再重新import一遍

即,添加一行代码:(即使之前import 过torch ,也要在手动import一下torch中的utils.data)

import torch.utils.data

原因:因为torch.utils.data中的data需要手动的引入一遍。而且,有的包是不能直接 . 出来的,就得手动导入

参考链接:

1. Torch.utils has no module 'data' - vision - PyTorch Forums

2. https://www.jianshu.com/p/4db9ff5b7485

3. AttributeError: module 'torch.utils' has no attribute 'data' 解决方法_rocketeerLi的博客-CSDN博客

你可能感兴趣的:(#,Pytorch学习,pycharm,pytorch)