hparams

hyperparams.yaml

@@ -73,12 +73,6 @@ seq_lin: !new:speechbrain.nnet.linear.Linear
     input_size: !ref <d_model>
     n_neurons: !ref <output_neurons>
 
-asr_encoder: !new:torch.nn.ModuleList
-    - [!ref <CNN>, !ref <Transformer>]
-
-asr_model: !new:torch.nn.ModuleList
-    - [!ref <CNN>, !ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
-
 beam_searcher: !new:speechbrain.decoders.S2STransformerBeamSearch
     modules: [!ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
     bos_index: !ref <bos_index>
@@ -124,8 +118,12 @@ lm_model: !new:speechbrain.lobes.models.transformer.TransformerLM.TransformerLM
 
 tokenizer: !new:sentencepiece.SentencePieceProcessor
 
+# Models
+asr_encoder: !new:torch.nn.ModuleList
+    - [!ref <CNN>, !ref <Transformer>]
+
 asr_model: !new:torch.nn.ModuleList
-   - [!ref <enc>, !ref <emb>, !ref <dec>, !ref <ctc_lin>, !ref <seq_lin>]
+    - [!ref <CNN>, !ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
 
 modules:
    compute_features: !ref <compute_features>