Image normalization

Usually, the images are scaled to the [0, 1] interval first (images = images / 255). Then, to normalize them, you can use torchvision's transforms.Normalize and this is a link to the post where it’s explained how to compute the mean and std of the images.