--- a
+++ b/bin/train_bac_3d.sh
@@ -0,0 +1,27 @@
+#!/usr/bin/env bash
+
+export CUDA_VISIBLE_DEVICES=0,1
+RUN_CONFIG=config_3d.yml
+
+
+for model_name in resnet50; do
+    WANDB=1
+    for fold in 0 1 2 3 4; do
+        # Train and test csv
+        train_csv=./csv/patient2_kfold/train_$fold.csv
+        valid_csv=./csv/patient2_kfold/valid_$fold.csv
+
+        #stage 1
+        log_name=${model_name}-anjuu-512-$fold
+        LOGDIR=/logs/rsna/test/${log_name}/
+        USE_WANDB=${WANDB} catalyst-dl run \
+            --config=./configs/${RUN_CONFIG} \
+            --logdir=$LOGDIR \
+            --out_dir=$LOGDIR:str \
+            --monitoring_params/name=${log_name}:str \
+            --model_params/model_name=${model_name}:str \
+            --stages/data_params/train_csv=${train_csv}:str \
+            --stages/data_params/valid_csv=${valid_csv}:str \
+            --verbose
+    done
+done
\ No newline at end of file