Titouan commited on
Commit
0a69b53
1 Parent(s): e51c900

update interface

Browse files

Files changed (1) hide show
  1. 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
- pre_transformer: !ref <CNN>
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>