跑手写数字识别代码时，出现 NameError：name ‘transform‘ is not defined错误

跑手写数字识别代码时，出现 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])
               ]))