Titouan
commited on
Commit
·
0a69b53
1
Parent(s):
e51c900
update interface
Browse files- hyperparams.yaml +5 -2
hyperparams.yaml
CHANGED
@@ -118,14 +118,17 @@ lm_model: !new:speechbrain.lobes.models.transformer.TransformerLM.TransformerLM
|
|
118 |
|
119 |
tokenizer: !new:sentencepiece.SentencePieceProcessor
|
120 |
|
|
|
|
|
|
|
|
|
121 |
# Models
|
122 |
asr_model: !new:torch.nn.ModuleList
|
123 |
- [!ref <CNN>, !ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
|
124 |
|
125 |
modules:
|
126 |
compute_features: !ref <compute_features>
|
127 |
-
|
128 |
-
transformer: !ref <Transformer>
|
129 |
asr_model: !ref <asr_model>
|
130 |
normalize: !ref <normalize>
|
131 |
lm_model: !ref <lm_model>
|
|
|
118 |
|
119 |
tokenizer: !new:sentencepiece.SentencePieceProcessor
|
120 |
|
121 |
+
asr_encoder: !speechbrain.utils.callchains.LengthsCapableChain
|
122 |
+
- !ref <CNN>
|
123 |
+
- !ref <Transformer.encode>
|
124 |
+
|
125 |
# Models
|
126 |
asr_model: !new:torch.nn.ModuleList
|
127 |
- [!ref <CNN>, !ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
|
128 |
|
129 |
modules:
|
130 |
compute_features: !ref <compute_features>
|
131 |
+
asr_encoder: !ref <asr_encoder>
|
|
|
132 |
asr_model: !ref <asr_model>
|
133 |
normalize: !ref <normalize>
|
134 |
lm_model: !ref <lm_model>
|