Adel-Moumen commited on
Commit
0e7fdb9
1 Parent(s): 107e64f

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +30 -19
hyperparams.yaml CHANGED
@@ -3,7 +3,7 @@
3
  # Encoder: CRDNN model
4
  # Decoder: GRU + beamsearch + RNNLM
5
  # Tokens: BPE with unigram
6
- # Authors: Ju-Chieh Chou, Mirco Ravanelli, Abdel Heba, Peter Plantinga 2020
7
  # ############################################################################
8
 
9
 
@@ -124,24 +124,35 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
124
  normalize: !ref <normalizer>
125
  model: !ref <enc>
126
 
127
- decoder: !new:speechbrain.decoders.S2SRNNBeamSearchLM
128
- embedding: !ref <emb>
129
- decoder: !ref <dec>
130
- linear: !ref <seq_lin>
131
- language_model: !ref <lm_model>
132
- bos_index: !ref <bos_index>
133
- eos_index: !ref <eos_index>
134
- min_decode_ratio: !ref <min_decode_ratio>
135
- max_decode_ratio: !ref <max_decode_ratio>
136
- beam_size: !ref <beam_size>
137
- eos_threshold: !ref <eos_threshold>
138
- using_max_attn_shift: !ref <using_max_attn_shift>
139
- max_attn_shift: !ref <max_attn_shift>
140
- coverage_penalty: !ref <coverage_penalty>
141
- lm_weight: !ref <lm_weight>
142
- temperature: !ref <temperature>
143
- temperature_lm: !ref <temperature_lm>
144
-
 
 
 
 
 
 
 
 
 
 
 
145
 
146
  modules:
147
  normalizer: !ref <normalizer>
 
3
  # Encoder: CRDNN model
4
  # Decoder: GRU + beamsearch + RNNLM
5
  # Tokens: BPE with unigram
6
+ # Authors: Ju-Chieh Chou, Mirco Ravanelli, Abdel Heba, Peter Plantinga 2020, Adel Moumen 2024
7
  # ############################################################################
8
 
9
 
 
124
  normalize: !ref <normalizer>
125
  model: !ref <enc>
126
 
127
+ # Scorer
128
+ coverage_scorer: !new:speechbrain.decoders.scorer.CoverageScorer
129
+ vocab_size: !ref <output_neurons>
130
+
131
+ rnnlm_scorer: !new:speechbrain.decoders.scorer.RNNLMScorer
132
+ language_model: !ref <lm_model>
133
+ temperature: !ref <temperature_lm>
134
+
135
+ scorer: !new:speechbrain.decoders.scorer.ScorerBuilder
136
+ full_scorers: [!ref <rnnlm_scorer>,
137
+ !ref <coverage_scorer>]
138
+ weights:
139
+ rnnlm: !ref <lm_weight>
140
+ coverage: !ref <coverage_penalty>
141
+
142
+ decoder: !new:speechbrain.decoders.S2SRNNBeamSearcher
143
+ embedding: !ref <emb>
144
+ decoder: !ref <dec>
145
+ linear: !ref <seq_lin>
146
+ bos_index: !ref <bos_index>
147
+ eos_index: !ref <eos_index>
148
+ min_decode_ratio: !ref <min_decode_ratio>
149
+ max_decode_ratio: !ref <max_decode_ratio>
150
+ beam_size: !ref <beam_size>
151
+ eos_threshold: !ref <eos_threshold>
152
+ using_max_attn_shift: !ref <using_max_attn_shift>
153
+ max_attn_shift: !ref <max_attn_shift>
154
+ temperature: !ref <temperature>
155
+ scorer: !ref <scorer>
156
 
157
  modules:
158
  normalizer: !ref <normalizer>