AutoEncoder一节中的transform错误

[tanjuntao]

img_transform = transforms.Compose([
    transforms.ToTensor(),
    transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
])

这里有问题，我按照别人其他代码里面的transform改成下面这样可以跑

img_transform= transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,))])

[LeiWang1999]

Yes, 我提了个PR，已经合并了。