Every dataset is normalized with MNIST's mean and std dev

[npnbpb]

The transform used while loading every dataset (line 37, main.py) uses MNIST's mean and standard deviation. The correct values for normalizing CIFAR 10, etc are different.

[bainro]

For cifar10 & cifar100:
normalize = transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])