Titouan commited on
Commit
ec90d09
1 Parent(s): fab9c20

EncoderDecoderASR update

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +10 -8
hyperparams.yaml CHANGED
@@ -117,7 +117,14 @@ tokenizer: !new:sentencepiece.SentencePieceProcessor
117
  asr_model: !new:torch.nn.ModuleList
118
  - [!ref <enc>, !ref <emb>, !ref <dec>, !ref <ctc_lin>, !ref <seq_lin>]
119
 
120
- beam_searcher: !new:speechbrain.decoders.S2SRNNBeamSearchLM
 
 
 
 
 
 
 
121
  embedding: !ref <emb>
122
  decoder: !ref <dec>
123
  linear: !ref <seq_lin>
@@ -137,14 +144,9 @@ beam_searcher: !new:speechbrain.decoders.S2SRNNBeamSearchLM
137
 
138
 
139
  modules:
140
- compute_features: !ref <compute_features>
141
- normalize: !ref <normalize>
142
- asr_model: !ref <asr_model>
143
- asr_encoder: !ref <enc>
144
- asr_decoder: !ref <dec>
145
  lm_model: !ref <lm_model>
146
- beam_searcher: !ref <beam_searcher>
147
-
148
 
149
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
150
  loadables:
 
117
  asr_model: !new:torch.nn.ModuleList
118
  - [!ref <enc>, !ref <emb>, !ref <dec>, !ref <ctc_lin>, !ref <seq_lin>]
119
 
120
+ # We compose the inference (encoder) pipeline.
121
+ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
122
+ input_shape: [null, null, !ref <n_mels>]
123
+ compute_features: !ref <compute_features>
124
+ normalize: !ref <normalize>
125
+ model: !ref <enc>
126
+
127
+ decoder: !new:speechbrain.decoders.S2SRNNBeamSearchLM
128
  embedding: !ref <emb>
129
  decoder: !ref <dec>
130
  linear: !ref <seq_lin>
 
144
 
145
 
146
  modules:
147
+ encoder: !ref <encoder>
148
+ decoder: !ref <decoder>
 
 
 
149
  lm_model: !ref <lm_model>
 
 
150
 
151
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
152
  loadables: