I had same issue, then adding brackets to the mean and std value inside transform.Normalize fixed this.
As @qi_chang replied this should work:



 qi_chang:

transform_list = [transforms.ToTensor(),
transforms.Normalize([0.5], [0.5])]