So I work at a transfer learning example. My task is to take VGG-16, modify and train it using transfer learning only to predict 2 labels ("Cat" and "Dog"