ERROR PyTorch 0.1.11
It’s hard to read the code in my phone.
maybe you can use markdown to beautify your code.
and you need
also make sure you have the same preprocessing for images during training and validate period
Sorry about the styling.
But adding net.eval() worked great. Now it classifies correctly.
Thanks for the help!