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.
morowenka/simpsons-classification
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|