Mr.Crab:

However, the those accuracy is lower than 96.7% when is ues the pytorch code to do the inference.


What is the difference between the posted code snippet and the other code you are using to achieve the 96% accuracy?