-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.sh
19 lines (19 loc) · 796 Bytes
/
run.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#!/bin/bash
exp_type="$1" # exp_type: no-ema_no-aug, no-ema_aug, ema_aug
exp_num="$2" # exp_num: 1, 2, 3, 4, 5
shot_num="$3" # shot_num: 1, 5, 10
data_type="$4" # data_type: chest, colon, endo
config_file=configs/eva-1b_adapt_"$exp_type"_exp"$exp_num"/in21k-eva1b_adapt_bs4_lr6e-4_"$shot_num"-shot_"$data_type".py
if test -z "$exp_type" || test -z "$exp_num" || test -z "$shot_num" || test -z "$data_type"
then
echo "Missing parameter"
exit
else
echo start on $(date)
export PYTHONPATH=$PWD:$PYTHONPATH
export OMP_NUM_THREADS=12
export MKL_NUM_THREADS=12
echo Using configuration file: "$config_file"
python tools/train.pyc "$config_file" --gpu-ids 0 1 2 3 --local_rank 0 --seed 42 --deterministic
echo end on $(date)
fi