Screenshot from 2018-07-14 22-36-37.png1920×1080 285 KB

Well, my code is like above,  when I run mnist.py, there is an error:
transform = transforms.Compose(transforms.ToTensor(),transforms.Normalize(mean=[0.5], std=[0.5]))

TypeError: init() takes exactly 2 arguments (3 given)
However, I found the init function of class Normalize(object) have 2 inputs indeed, and they are std and mean. So could anyone can help to find what’s wrong with the codes?