-
Notifications
You must be signed in to change notification settings - Fork 7
/
train.sh
128 lines (123 loc) · 2.59 KB
/
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
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#!/bin/bash
gpu_id=0
nb_worker=24
save_dir = [SET_YOUR_PATH, ex)/result/DCASENET/]
db_tag = [SET_YOUR_PATH, ex)/data/DCASE2019_Task2/]
db_asc = [SET_YOUR_PATH, ex)/data/DCASE2020_Task1A/]
db_sed = [SET_YOUR_PATH, ex)/data/DCASE2020_Task3/]
#########
# Joint
#########
CUDA_VISIBLE_DEVICES=${gpu_id} python main.py -verbose 0 \
-nb_iter_per_epoch 500 \
-task ASC TAG SED \
-name Joint \
-model_scp DcaseNet \
-model_name get_DcaseNet_v3 \
-nb_worker ${nb_worker} \
-wd 0.0000\
-epoch 160 \
-bs_ASC 32 \
-bs_SED 24 \
-bs_TAG 32 \
-optimizer adam \
-amsgrad 0 \
-do_lr_decay 1 \
-lr_decay cosine \
-lrdec_t0 80 \
-do_mixup 1 \
-mixup_start 5 \
-loss_weight_ASC 1 \
-loss_weight_SED 1 \
-loss_weight_TAG 1 \
-save_dir ${save_dir} \
-DB_TAG ${db_tag} \
-DB_ASC ${db_asc} \
-DB_SED ${db_sed}
#########
# fine-tuneASC
#########
CUDA_VISIBLE_DEVICES=${gpu_id} python main.py -verbose 0 \
-nb_iter_per_epoch 500 \
-task ASC\
-name fine-tuneASC \
-model_scp DcaseNet \
-model_name get_DcaseNet_v3 \
-nb_worker ${nb_worker} \
-wd 0.0000\
-epoch 160 \
-bs_ASC 32 \
-bs_SED 24 \
-bs_TAG 32 \
-optimizer adam \
-amsgrad 0 \
-do_lr_decay 1 \
-lr_decay cosine \
-lrdec_t0 80 \
-do_mixup 1 \
-mixup_start 5 \
-loss_weight_ASC 1 \
-loss_weight_SED 1 \
-loss_weight_TAG 1 \
-save_dir ${save_dir} \
-DB_TAG ${db_tag} \
-DB_ASC ${db_asc} \
-DB_SED ${db_sed}
#########
# fine-tuneTAG
#########
CUDA_VISIBLE_DEVICES=${gpu_id} python main.py -verbose 0 \
-nb_iter_per_epoch 500 \
-task TAG\
-name fine-tuneTAG \
-model_scp DcaseNet \
-model_name get_DcaseNet_v3 \
-nb_worker ${nb_worker} \
-wd 0.0000\
-epoch 160 \
-bs_ASC 32 \
-bs_SED 24 \
-bs_TAG 32 \
-optimizer adam \
-amsgrad 0 \
-do_lr_decay 1 \
-lr_decay cosine \
-lrdec_t0 80 \
-do_mixup 1 \
-mixup_start 5 \
-loss_weight_ASC 1 \
-loss_weight_SED 1 \
-loss_weight_TAG 1 \
-save_dir ${save_dir} \
-DB_TAG ${db_tag} \
-DB_ASC ${db_asc} \
-DB_SED ${db_sed}
#########
# fine-tuneSED
#########
CUDA_VISIBLE_DEVICES=${gpu_id} python main.py -verbose 0 \
-nb_iter_per_epoch 500 \
-task SED\
-name fine-tuneSED \
-model_scp DcaseNet \
-model_name get_DcaseNet_v3 \
-nb_worker ${nb_worker} \
-wd 0.0000\
-epoch 160 \
-bs_ASC 32 \
-bs_SED 24 \
-bs_TAG 32 \
-optimizer adam \
-amsgrad 0 \
-do_lr_decay 1 \
-lr_decay cosine \
-lrdec_t0 80 \
-do_mixup 1 \
-mixup_start 5 \
-loss_weight_ASC 1 \
-loss_weight_SED 1 \
-loss_weight_TAG 1 \
-save_dir ${save_dir} \
-DB_TAG ${db_tag} \
-DB_ASC ${db_asc} \
-DB_SED ${db_sed}