- torch == 1.1.0
- cnn_finetune == 0.5.3
- albumentations == 0.2.3
- catalyst == 19.06.5
-
In
configs/config.yml
Change:stages/data_params/root
to your root data that you download from Kaggle -
In
bin/train.sh
Change:export CUDA_VISIBLE_DEVICES=2,3
which GPUs are used for trainingLOGDIR
: saved checkpoints, logs, etc
-
Run:
bash bin/train.sh
-
Take a coffe or go to sleep
Please refer make_submission.py
.
- Change following:
test_csv = '/raid/data/kaggle/recursion-cellular-image-classification/test.csv'
# test_csv = './csv/valid_0.csv'
log_dir = "/raid/bac/kaggle/logs/recursion_cell/test/rgb_no_crop_512_accum2_456/se_resnext50_32x4d/"
root = "/raid/data/kaggle/recursion-cellular-image-classification/"
- Run:
python src/make_submission.csv