ianweb100:

transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225] )])


It turns out the issue is on the transforms normalise, i was using 3 channels for a 1 channel data. I now manage to easily load, shuffle and reload the data using imageFolder ( Thanks to Patrick ) and some other techniques I picked here in the forum.
Thanks for all the help.