-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathjoin_train.sh
26 lines (26 loc) · 848 Bytes
/
join_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
python joint_train.py \
--train-manifest-list data/manifests/cv-valid-train_manifest.csv data/manifests/hkust_16khz_train_manifest.csv data/manifests/seame_phaseII_train_manifest.csv \
--valid-manifest-list data/manifests/cv-valid-dev_manifest.csv data/manifests/hkust_16khz_dev_manifest.csv data/manifests/seame_phaseII_val_manifest.csv \
--cuda --k-train 8 \
--labels-path data/labels/hkust_seame_labels.json \
--lr 1e-4 \
--name joint_enc2_dec4_512_b8_16khz \
--save-every 10000 \
--feat_extractor vgg_cnn \
--dropout 0.1 \
--num-enc-layers 2 \
--num-dec-layers 4 \
--num-heads 8 \
--dim-model 512 \
--dim-key 64 \
--dim-value 64 \
--dim-input 5120 \
--dim-inner 512 \
--dim-emb 512 \
--early-stop cer,200 \
--src-max-len 5000 \
--tgt-max-len 2500 \
--evaluate-every 10000 \
--epochs 10000000 \
--sample-rate 16000 \
--train-partition-list 1 1 1