В данном ноутбуке рассматривается решение задачи классификации изображений в рамках учебного Kaggle-соревнования Journey to Springfield от Deep Learning School. Для решения используется техника Transfer Learning/Fine tuning над нейронными сетями Efficientnet_B4
и ResNet 18
, предобученными на ImageNet. В этом блокноте приведён основной training loop, выполняется аугментация изображений, в качестве оптимизатора используется AdamW. Для работы с learning rate используется scheduler StepLR
. Для борьбы с дисбалансом классов используется oversampling.
Результат на Kaggle: 0.99362 F1-Score
.