Shouldn’t the mean and std 18 be tuples?

normalize = torchvision.transforms.Normalize(mean=(0.485, 0.456, 0.406), std=(0.229, 0.224, 0.225) )

Or, it is OK to use list sequences as shown in your case?