Titouan commited on
Commit
8279335
1 Parent(s): 63e5d51

new interface for EncoderDecoderASR

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +10 -7
hyperparams.yaml CHANGED
@@ -105,7 +105,14 @@ asr_model: !new:torch.nn.ModuleList
105
 
106
  tokenizer: !new:sentencepiece.SentencePieceProcessor
107
 
108
- beam_searcher: !new:speechbrain.decoders.S2SRNNBeamSearcher
 
 
 
 
 
 
 
109
  embedding: !ref <emb>
110
  decoder: !ref <dec>
111
  linear: !ref <seq_lin>
@@ -120,12 +127,8 @@ beam_searcher: !new:speechbrain.decoders.S2SRNNBeamSearcher
120
  temperature: !ref <temperature>
121
 
122
  modules:
123
- compute_features: !ref <compute_features>
124
- normalize: !ref <normalize>
125
- asr_model: !ref <asr_model>
126
- asr_encoder: !ref <enc>
127
- asr_decoder: !ref <dec>
128
- beam_searcher: !ref <beam_searcher>
129
 
130
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
131
  loadables:
105
 
106
  tokenizer: !new:sentencepiece.SentencePieceProcessor
107
 
108
+ # We compose the inference (encoder) pipeline.
109
+ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
110
+ input_shape: [null, null, !ref <n_mels>]
111
+ compute_features: !ref <compute_features>
112
+ normalize: !ref <normalize>
113
+ model: !ref <enc>
114
+
115
+ decoder: !new:speechbrain.decoders.S2SRNNBeamSearcher
116
  embedding: !ref <emb>
117
  decoder: !ref <dec>
118
  linear: !ref <seq_lin>
127
  temperature: !ref <temperature>
128
 
129
  modules:
130
+ encoder: !ref <encoder>
131
+ decoder: !ref <decoder>
 
 
 
 
132
 
133
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
134
  loadables: