-
Notifications
You must be signed in to change notification settings - Fork 0
/
train.sh
45 lines (35 loc) · 1018 Bytes
/
train.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
task=$1
model=$2
modelName=$3
encoderName=$4
learnSize=$5
gpuN=$6
folds=("0" "1" "2" "3" "4")
if [[ $task =~ ^(ISIC)$ ]]; then
for i in ${folds[@]}; do
echo "==================================================="
echo "fold : $i/4"
echo "task : $task"
echo "model : $model"
echo "modelName : $modelName"
echo "encoderName : $encoderName"
echo "==================================================="
python3 ./train.py \
--task $task \
--model $model \
--modelName $modelName \
--encoderName $encoderName \
--learnSize $learnSize \
--fold $i \
--gpuN $gpuN
done
else
python3 ./train.py \
--task $task \
--model $model \
--modelName $modelName\
--encoderName $encoderName \
--learnSize $learnSize \
--fold 0 \
--gpuN $gpuN
fi;