Titouan
commited on
Commit
·
e51c900
1
Parent(s):
2c62b42
remove transformer interface
Browse files- hyperparams.yaml +2 -5
hyperparams.yaml
CHANGED
@@ -122,13 +122,10 @@ tokenizer: !new:sentencepiece.SentencePieceProcessor
|
|
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 |
-
|
|
|
132 |
asr_model: !ref <asr_model>
|
133 |
normalize: !ref <normalize>
|
134 |
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 |
modules:
|
126 |
compute_features: !ref <compute_features>
|
127 |
+
pre_transformer: !ref <CNN>
|
128 |
+
transformer: !ref <Transformer>
|
129 |
asr_model: !ref <asr_model>
|
130 |
normalize: !ref <normalize>
|
131 |
lm_model: !ref <lm_model>
|