Tejus

@bignightcat Replace ``` # data_loader img_size = 64 transform = transforms.Compose([ transforms.Scale(img_size), transforms.ToTensor(), transforms.Normalize(mean=(0.5,0.5,0.5), std=(0.5,0.5,0.5)) ]) ``` with ``` # data_loader img_size = 64 transform = transforms.Compose([ transforms.Scale(img_size), transforms.ToTensor(), transforms.Normalize(mean=[0.5],...

I would like to work on this issue!