Solved by TinfoilHat0 in post #4 
              
              
                Remove normalizer from test transforms. 
This might reduce your test accuracy though. Ideally, you’d want to unnormalize images at the time you plot them.