simpsons-classification Here I have written a model and a pre-trained model for a classification task. I have implemented the train loop by myself. The model for fine-tuning was an EfficientNet B6.