Java Vs Python Image Classifier In Pytorch Cuda