Binary Image Classification Model Tensorflow Vs Pytorch