pytorch_ssd_active_learning/generate_plots.bash at master · CtfChan/pytorch_ssd_active_learning · GitHub

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
#!/bin/bash

# with weight reset
mkdir -p results/no_weight_reinit
for value in {1..3}
do
    for af in QBC RANDOM MEAN_STD ENTROPY MEAN_STD_WITH_BBOX LOCALIZATION_STABILITY VAR_RATIO MARGIN_SAMPLING BALD
    do
       python3 active_train.py --trial_number $value --acquisition_function $af --lr 1e-03 \
                --reset_weight False --save_dir ./results/no_weight_reinit/
    done
done


# with weight reset
mkdir -p results/weight_reinit
for value in {1..3}
do
    for af in QBC RANDOM MEAN_STD ENTROPY MEAN_STD_WITH_BBOX LOCALIZATION_STABILITY VAR_RATIO MARGIN_SAMPLING BALD
    do
        python3 active_train.py --trial_number $value --acquisition_function $af --lr 1e-03 \
                --reset_weight True --save_dir ./results/weight_reinit/
    done
done


echo All done