I am using transfer learning technique on Alexnet and other models along with image augmentation using pytorch lightning. The f1 scores look good, as I would expect, the val