Cannot find reference ‘data‘ in ‘__init__.py‘——在pycharm中调用torch.utils.data.DataLoader()提示Warning

PyTorch1.7中文官方教程《使用 TensorBoard 可视化模型,数据和训练》一节,使用DataLoader代码:
调用DataLoader(),data下划线Warning
data处有下划线Warning:

Cannot find reference 'data' in '__init__.py' 

消除warning方法:

  • 将 data模块 import 进程序:
import torch.utils.data
  • 直接import DataLoader,代码中直接实例化 DataLoader
from torch.utils.data import DataLoader
trainloader = DataLoader(trainset, batch_size=4,
                                        shuffle=True, num_workers=2)

testloader = DataLoader(testset, batch_size=4,
                                        shuffle=False, num_workers=2)

猜你喜欢

转载自blog.csdn.net/weixin_43196818/article/details/114948229