Titouan
commited on
Commit
•
ec90d09
1
Parent(s):
fab9c20
EncoderDecoderASR update
Browse files- hyperparams.yaml +10 -8
hyperparams.yaml
CHANGED
@@ -117,7 +117,14 @@ tokenizer: !new:sentencepiece.SentencePieceProcessor
|
|
117 |
asr_model: !new:torch.nn.ModuleList
|
118 |
- [!ref <enc>, !ref <emb>, !ref <dec>, !ref <ctc_lin>, !ref <seq_lin>]
|
119 |
|
120 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
121 |
embedding: !ref <emb>
|
122 |
decoder: !ref <dec>
|
123 |
linear: !ref <seq_lin>
|
@@ -137,14 +144,9 @@ beam_searcher: !new:speechbrain.decoders.S2SRNNBeamSearchLM
|
|
137 |
|
138 |
|
139 |
modules:
|
140 |
-
|
141 |
-
|
142 |
-
asr_model: !ref <asr_model>
|
143 |
-
asr_encoder: !ref <enc>
|
144 |
-
asr_decoder: !ref <dec>
|
145 |
lm_model: !ref <lm_model>
|
146 |
-
beam_searcher: !ref <beam_searcher>
|
147 |
-
|
148 |
|
149 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
150 |
loadables:
|
|
|
117 |
asr_model: !new:torch.nn.ModuleList
|
118 |
- [!ref <enc>, !ref <emb>, !ref <dec>, !ref <ctc_lin>, !ref <seq_lin>]
|
119 |
|
120 |
+
# We compose the inference (encoder) pipeline.
|
121 |
+
encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
122 |
+
input_shape: [null, null, !ref <n_mels>]
|
123 |
+
compute_features: !ref <compute_features>
|
124 |
+
normalize: !ref <normalize>
|
125 |
+
model: !ref <enc>
|
126 |
+
|
127 |
+
decoder: !new:speechbrain.decoders.S2SRNNBeamSearchLM
|
128 |
embedding: !ref <emb>
|
129 |
decoder: !ref <dec>
|
130 |
linear: !ref <seq_lin>
|
|
|
144 |
|
145 |
|
146 |
modules:
|
147 |
+
encoder: !ref <encoder>
|
148 |
+
decoder: !ref <decoder>
|
|
|
|
|
|
|
149 |
lm_model: !ref <lm_model>
|
|
|
|
|
150 |
|
151 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
152 |
loadables:
|