|
a |
|
b/bin/train_toan.sh |
|
|
1 |
#!/usr/bin/env bash |
|
|
2 |
|
|
|
3 |
export CUDA_VISIBLE_DEVICES=6,7 |
|
|
4 |
RUN_CONFIG=config_toan.yml |
|
|
5 |
|
|
|
6 |
|
|
|
7 |
for fold in 0 1 2 3 4; do |
|
|
8 |
#stage 1 |
|
|
9 |
log_name=densenet169-mw-512-$fold |
|
|
10 |
LOGDIR=/logs/rsna/test/${log_name}/ |
|
|
11 |
catalyst-dl run \ |
|
|
12 |
--config=./configs/${RUN_CONFIG} \ |
|
|
13 |
--logdir=$LOGDIR \ |
|
|
14 |
--out_dir=$LOGDIR:str \ |
|
|
15 |
--monitoring_params/name=${log_name}:str \ |
|
|
16 |
--stages/data_params/train_csv=./csv/patient2_kfold/train_$fold.csv:str \ |
|
|
17 |
--stages/data_params/valid_csv=./csv/patient2_kfold/valid_$fold.csv:str \ |
|
|
18 |
--verbose |
|
|
19 |
done |