Skip to content

arkravchuk/kaggla-alaska

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Код решения (20 место) kaggle ALASKA2 Image Steganalysis

Решение (фитпредикт, ничего умного):

  • Блендинг 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 гпу часов.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages