[RESOLVED] Image Classifier Not Predicting Correctly

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
before output=net(image)

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!