-
Notifications
You must be signed in to change notification settings - Fork 1
/
config.yaml
54 lines (45 loc) · 1.35 KB
/
config.yaml
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
data:
data_root: "/home/xieliang/Data"
trained_transducer: "/home/xieliang/Data/transducer.pth"
trained_ctc: "/home/xieliang/Data/ctc2_test.pth"
trained_lm: "/home/xieliang/Data/lm.pth"
trained_las: "/home/xieliang/Data/las.pth"
vocabSize: 41 # the vocab includes '<pad>' and '<blank>' tokens
blank_idx: 1 # the index of '<blank>' token in vocabulary
pad_idx: 0 # the index of '<pad>' token in vocabulary
using_trained_ctc: True
using_trained_lm: True
model:
ctc:
nhid: 256
nlayer: 1
sigma: 0.01 # noise item
lm:
nemd: 128
nhid: 128
nlayer: 1
tie_embedding: False
embed_drop_ratio: 0.2 # embedding regularize
lock_drope: 0.2 # embedding lock drop ratio
lock_droph: 0 # lstm hidden lock drop ratio
lock_dropo: 0.2 # lstm out lock drop ratio
transducer:
encoder_nout: 256
decoder_nout: 256
nhid: 256
las:
encoder:
nhid: 0
nlayer: 0
nout: 0
attention:
nhid: 0
decoder:
nemd: 0
nhid: 0
emd_drop: 0
training:
BSZ: 1
lr: 0.0001
momentum: 0.9
weight_decay: 0