--- a
+++ b/bin/multi_size.sh
@@ -0,0 +1,26 @@
+#!/usr/bin/env bash
+
+export CUDA_VISIBLE_DEVICES=0,1
+RUN_CONFIG=multi_size.yml
+
+
+model_name=alexnet
+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}-mww-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
\ No newline at end of file