Also, from what I can tell, your normalized images are getting thrown out.  That image on line 21 isn’t referenced again, and it would only contain the final image in any case, but I could be missing something in the code.  It’s pretty standard to separate out the transforms into a separate transforms.compose function and pass that into your dataloader so that you don’t have to worry about it in your training loop.  That might things easier to troubleshoot as well.