跑手写数字识别代码时,出现 NameError:name ‘transform’ is not defined错误
这个错误并不是因为代码开头没导入相关包,而是在
data_train = datasets.MNIST(
root="../data/",
transform=transform,
train=True,
download=True
)
data_test = datasets.MNIST(
root="../data/",
transform=transform,
train=False
)
前未设置transform
修改如下,加上:
transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize(mean=[0.5], std=[0.5])])
或者一步到位:
datasets.MNIST('../data/', train=True, download=True,
transform=transforms.Compose([
transforms.ToTensor(),
transforms.Normalize(mean=[0.5], std=[0.5])
]))