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>
|