源代码:
import torch
import torchvision
import torch.nn as nn
from model import LeNet
import torch.optim as optim
import torchvision.transforms as transforms
from torch.utils.data import DataLoader
transform = transforms.Compose(
[transforms.ToTensor(),
transforms.Normalize(0.5, 0.5,0.5), (0.5,0.5,0.5)])
以上是部分源代码,为展示完整代码。出现标题所示错误的原因是:
transforms.Normalize(0.5, 0.5,0.5), (0.5,0.5,0.5)
改为
transforms.Normalize((0.5, 0.5,0.5), (0.5,0.5,0.5))
即可
细节!细节!细节!