About 6 months ago, I used this transforms for MNIST datasets and It had been worked well.
normalize = transforms.Normalize(mean=[0.5, 0.5, 0.5],
                                 std=[0.5, 0.5, 0.5])

But now, above code have error, because MNIST have channel only 1 I think.
So I changed above codes  to this. and It works.
normalize = transforms.Normalize(mean=[0.5],
                                 std=[0.5])

Is there any update pytorch? or something I missing?