PyTorch1.7中文官方教程《使用 TensorBoard 可视化模型,数据和训练》一节,使用DataLoader代码:
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)