-
Notifications
You must be signed in to change notification settings - Fork 14
/
train_regdb.sh
78 lines (75 loc) · 1.73 KB
/
train_regdb.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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
array=(
"trial1"
"trial2"
"trial3"
"trial4"
"trial5"
"trial6"
"trial7"
"trial8"
"trial9"
"trial10"
)
# for idx in $( seq 1 10 )
# do
# CUDA_VISIBLE_DEVICES=0 python -W ignore -u train_base.py \
# --data_root "/path/to/your/RegDB/" \
# --dataset "regdb" \
# --save "./checkpoints/" \
# --print_freq 5 \
# --test_freq 10 \
# --epochs 120 \
# --batch_size 64 \
# --test_batch 128 \
# --num_pos 4 \
# --optim 'adam' \
# --lr 0.01 \
# --beta1 0.5 \
# --beta2 0.999 \
# --weight_decay 5e-4 \
# --img_w 128 \
# --img_h 256 \
# --label_smooth 0.0 \
# --last_stride 1 \
# --dropout_rate 0.0 \
# --ema_decay 0.0 \
# --sp_lambda 0.0 \
# --cmmd_lambda 0.0 \
# --margin 0.4 \
# --triplet_feat_norm 'no' \
# --test_feat_norm 'yes' \
# --trial $idx \
# --note 'regdb-baseline/'${array[$idx]}
# done
for idx in $( seq 1 10 )
do
CUDA_VISIBLE_DEVICES=0 python -W ignore -u train_eval.py \
--data_root "/path/to/your/RegDB/" \
--dataset "regdb" \
--save "./checkpoints/" \
--config_path "./checkpoints_regdb/cm-nas_searched_arch/config.cfg" \
--print_freq 5 \
--test_freq 10 \
--epochs 120 \
--batch_size 64 \
--test_batch 128 \
--num_pos 4 \
--optim 'adam' \
--lr 0.01 \
--beta1 0.5 \
--beta2 0.999 \
--weight_decay 5e-4 \
--img_w 128 \
--img_h 256 \
--label_smooth 0.0 \
--last_stride 1 \
--dropout_rate 0.0 \
--ema_decay 0.968 \
--sp_lambda 5.0 \
--cmmd_lambda 0.05 \
--margin 0.4 \
--triplet_feat_norm 'no' \
--test_feat_norm 'yes' \
--trial $idx \
--note 'regdb-eval-ema/'${array[$idx]}
done