Adel-Moumen
commited on
Commit
•
e9d297b
1
Parent(s):
9b2cedf
Update hyperparams.yaml
Browse files- hyperparams.yaml +9 -0
hyperparams.yaml
CHANGED
@@ -79,23 +79,30 @@ dec: !new:speechbrain.nnet.RNN.AttentionalRNNDecoder
|
|
79 |
kernel_size: 100
|
80 |
re_init: True
|
81 |
dropout: !ref <dropout>
|
|
|
82 |
ctc_lin: !new:speechbrain.nnet.linear.Linear
|
83 |
input_size: !ref <dnn_neurons>
|
84 |
n_neurons: !ref <output_neurons>
|
|
|
85 |
seq_lin: !new:speechbrain.nnet.linear.Linear
|
86 |
input_size: !ref <dec_neurons>
|
87 |
n_neurons: !ref <output_neurons>
|
|
|
88 |
log_softmax: !new:speechbrain.nnet.activations.Softmax
|
89 |
apply_log: True
|
|
|
90 |
asr_model: !new:torch.nn.ModuleList
|
91 |
- [!ref <enc>, !ref <emb>, !ref <dec>, !ref <ctc_lin>, !ref <seq_lin>]
|
|
|
92 |
tokenizer: !new:sentencepiece.SentencePieceProcessor
|
|
|
93 |
# We compose the inference (encoder) pipeline.
|
94 |
encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
95 |
input_shape: [null, null, !ref <n_mels>]
|
96 |
compute_features: !ref <compute_features>
|
97 |
normalize: !ref <normalizer>
|
98 |
model: !ref <enc>
|
|
|
99 |
decoder: !new:speechbrain.decoders.S2SRNNBeamSearcher
|
100 |
embedding: !ref <emb>
|
101 |
decoder: !ref <dec>
|
@@ -109,10 +116,12 @@ decoder: !new:speechbrain.decoders.S2SRNNBeamSearcher
|
|
109 |
using_max_attn_shift: !ref <using_max_attn_shift>
|
110 |
max_attn_shift: !ref <max_attn_shift>
|
111 |
temperature: !ref <temperature>
|
|
|
112 |
modules:
|
113 |
normalizer: !ref <normalizer>
|
114 |
encoder: !ref <encoder>
|
115 |
decoder: !ref <decoder>
|
|
|
116 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
117 |
loadables:
|
118 |
normalizer: !ref <normalizer>
|
|
|
79 |
kernel_size: 100
|
80 |
re_init: True
|
81 |
dropout: !ref <dropout>
|
82 |
+
|
83 |
ctc_lin: !new:speechbrain.nnet.linear.Linear
|
84 |
input_size: !ref <dnn_neurons>
|
85 |
n_neurons: !ref <output_neurons>
|
86 |
+
|
87 |
seq_lin: !new:speechbrain.nnet.linear.Linear
|
88 |
input_size: !ref <dec_neurons>
|
89 |
n_neurons: !ref <output_neurons>
|
90 |
+
|
91 |
log_softmax: !new:speechbrain.nnet.activations.Softmax
|
92 |
apply_log: True
|
93 |
+
|
94 |
asr_model: !new:torch.nn.ModuleList
|
95 |
- [!ref <enc>, !ref <emb>, !ref <dec>, !ref <ctc_lin>, !ref <seq_lin>]
|
96 |
+
|
97 |
tokenizer: !new:sentencepiece.SentencePieceProcessor
|
98 |
+
|
99 |
# We compose the inference (encoder) pipeline.
|
100 |
encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
101 |
input_shape: [null, null, !ref <n_mels>]
|
102 |
compute_features: !ref <compute_features>
|
103 |
normalize: !ref <normalizer>
|
104 |
model: !ref <enc>
|
105 |
+
|
106 |
decoder: !new:speechbrain.decoders.S2SRNNBeamSearcher
|
107 |
embedding: !ref <emb>
|
108 |
decoder: !ref <dec>
|
|
|
116 |
using_max_attn_shift: !ref <using_max_attn_shift>
|
117 |
max_attn_shift: !ref <max_attn_shift>
|
118 |
temperature: !ref <temperature>
|
119 |
+
|
120 |
modules:
|
121 |
normalizer: !ref <normalizer>
|
122 |
encoder: !ref <encoder>
|
123 |
decoder: !ref <decoder>
|
124 |
+
|
125 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
126 |
loadables:
|
127 |
normalizer: !ref <normalizer>
|