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

PyTorch1.7中文官方教程《使用 TensorBoard 可视化模型,数据和训练》一节,使用DataLoader代码:
Cannot find reference ‘data‘ in ‘__init__.py‘——在pycharm中调用torch.utils.data.DataLoader()提示Warning_第1张图片
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)

你可能感兴趣的:(PyThorch学习,pytorch,pycharm,python)