tattaka / hms-harmful-brain-activity-classification-public

HMS - Harmful Brain Activity Classification 4th place solution

Home Page:https://www.kaggle.com/competitions/hms-harmful-brain-activity-classification/discussion/492240

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

4th place solution (tattaka's part)

Requirements

16GB x 4 VRAM (trained on NVIDIA RTX A4000 x 4).

Environment

Use Kaggle Docker.
Follow tattaka/ml_environment to build the environment.
You can run ./RUN-KAGGLE-GPU-ENV.sh and launch the docker container.

Usage

  1. Place competition data in the input directory
  2. Training
    $ cd src/exp092 && sh train.sh
    $ cd src/exp094 && sh train.sh
    $ cd src/exp107 && sh train.sh
    $ cd src/exp108 && sh train.sh
    $ cd src/exp146 && sh train.sh
    $ cd src/exp147 && sh train.sh

License

MIT

About

HMS - Harmful Brain Activity Classification 4th place solution

https://www.kaggle.com/competitions/hms-harmful-brain-activity-classification/discussion/492240

License:MIT License


Languages

Language:Python 75.4%Language:Shell 24.6%