Python Image Classifier In Pytorch Vs Tensorflow