在训练手写数字识别模型时,说我调用的方法中传入参数不对,python基础不是太好,看不出来哪里不对,求指点。
怀疑是调用 MNIST 数据集时,transform
参数里面的 Normalize
变换层没设置好,其中的 std 值被当作 0,造成除零问题。建议检查 MNIST 加载的选项,例如:
train_loader = torch.utils.data.DataLoader(
datasets.MNIST('../mnist_data',
download=True,
train=True,
transform=transforms.Compose([
transforms.ToTensor(), # first, convert image to PyTorch tensor
transforms.Normalize((0.1307,), (0.3081,)) # normalize inputs
])),
batch_size=10,
shuffle=True
)
你把for index,(data,label) in __:这里的元组改下试试