- Блендинг effnet b2, b3 с двух фолдов (к сожалению, после окончания соревнования по привычке нажал rm -rf на папку с чекпоинтами, поэтому не могу их приложить:( )
- Аугментации: hflip, wflip, random_rotate90
- Лосс: CrossEntropyLoss
- Оптимизатор: AdamW
В src/config.py
необходимо поставить соответствующие параметры для Вашей тренировки, а затем сделать
cd src
python train.py
В src
создастся папка experiments
, а в ней создастся папка с названием эксперимента. В этой папке будет храниться код эксперимента и лучший чекпоинт.
В src/inference_config.py
необходимо поставить соответствующие параметры для инференса, а затем сделать
cd src
python inference.py
Эксперименты проводились на TeslaV100. Каждая из моделей тренировалась ~трое суток, т.е. всего ~288 гпу часов.