--- a +++ b/bin/train_toan.sh @@ -0,0 +1,19 @@ +#!/usr/bin/env bash + +export CUDA_VISIBLE_DEVICES=6,7 +RUN_CONFIG=config_toan.yml + + +for fold in 0 1 2 3 4; do + #stage 1 + log_name=densenet169-mw-512-$fold + LOGDIR=/logs/rsna/test/${log_name}/ + catalyst-dl run \ + --config=./configs/${RUN_CONFIG} \ + --logdir=$LOGDIR \ + --out_dir=$LOGDIR:str \ + --monitoring_params/name=${log_name}:str \ + --stages/data_params/train_csv=./csv/patient2_kfold/train_$fold.csv:str \ + --stages/data_params/valid_csv=./csv/patient2_kfold/valid_$fold.csv:str \ + --verbose +done \ No newline at end of file