Binary Image Classification Model Tensorflow Vs Torch