-
Notifications
You must be signed in to change notification settings - Fork 16
/
Copy pathtrain_model.sh
29 lines (29 loc) · 1022 Bytes
/
train_model.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
#the split to train
SPLIT=(1)
SHOT=(1 2 3 5 10)
Test_weights=(01 02 03 04 05 06 07 08 09 10)
##base training
mkdir backup
for split in ${SPLIT[*]}
do
python tool/train_decoupling_disturbance.py cfg/fewshot/metayolo_split${split}.data cfg/darknet_dynamic.cfg cfg/reweighting_net_decoupling.cfg darknet19_448.conv.23 backup/split${split}_base
done
## finetune
for shot in ${SHOT[*]}
do
for split in ${SPLIT[*]}
do
python tool/train_decoupling_disturbance.py cfg/fewshot/metatune_${shot}shot_split${split}.data cfg/darknet_dynamic.cfg cfg/reweighting_net_decoupling.cfg backup/split${split}_base/000350.weights backup/split${split}_${shot}shot
done
done
## test weight
for shot in ${SHOT[*]}
do
for split in ${SPLIT[*]}
do
for test_weights in ${Test_weights[*]}
do
python tool/valid_decoupling.py cfg/fewshot/metatune_${shot}shot_split${split}.data cfg/darknet_dynamic.cfg cfg/reweighting_net_decoupling.cfg backup/split${split}_${shot}shot/0000${test_weights}.weights
done
done
done