自己使用的Pytroch版本 (1.1.0)
看pytorc版本的代码为
print(torch.__version__)
解决问题的办法
torch.utils.data.Dataloader
改为
torch.utils.data.dataloader
即把大写的D改为小写的d
加载的包为
from __future__ import print_function
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
import torchvision
from torchvision import datasets, transforms
import matplotlib.pyplot as plt
import numpy as np