Titouan
commited on
Commit
·
2c62b42
1
Parent(s):
a6de363
remove transformer interface
Browse files- hyperparams.yaml +5 -2
hyperparams.yaml
CHANGED
@@ -122,10 +122,13 @@ tokenizer: !new:sentencepiece.SentencePieceProcessor
|
|
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>
|
|
|
122 |
asr_model: !new:torch.nn.ModuleList
|
123 |
- [!ref <CNN>, !ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
|
124 |
|
125 |
+
asr_encoder: !new:speechbrain.nnet.containers.Sequential
|
126 |
+
- !ref <CNN>
|
127 |
+
- !ref <Transformer.encode>
|
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>
|