Java Vs Python Image Classifier In Pytorch