Adel-Moumen commited on
Commit
c3eaae3
1 Parent(s): 8dbfc21

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +17 -6
hyperparams.yaml CHANGED
@@ -119,17 +119,27 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
119
  cnn: !ref <CNN>
120
  transformer_encoder: !ref <Tencoder>
121
 
122
- decoder: !new:speechbrain.decoders.S2STransformerBeamSearch
123
- modules: [!ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
124
  bos_index: !ref <bos_index>
125
  eos_index: !ref <eos_index>
126
- blank_index: !ref <blank_index>
127
  min_decode_ratio: !ref <min_decode_ratio>
128
  max_decode_ratio: !ref <max_decode_ratio>
129
  beam_size: !ref <test_beam_size>
130
- ctc_weight: !ref <ctc_weight_decode>
131
- lm_weight: !ref <lm_weight>
132
- lm_modules: !ref <lm_model>
133
  temperature: !ref <temperature>
134
  temperature_lm: !ref <temperature_lm>
135
  using_eos_threshold: !ref <using_eos_threshold>
@@ -137,6 +147,7 @@ decoder: !new:speechbrain.decoders.S2STransformerBeamSearch
137
  length_normalization: !ref <length_normalization>
138
  using_max_attn_shift: !ref <using_max_attn_shift>
139
  max_attn_shift: !ref <max_attn_shift>
 
140
 
141
  modules:
142
  compute_features: !ref <compute_features>
 
119
  cnn: !ref <CNN>
120
  transformer_encoder: !ref <Tencoder>
121
 
122
+ transformerlm_scorer: !new:speechbrain.decoders.scorer.TransformerLMScorer
123
+ language_model: !ref <lm_model>
124
+
125
+ ctc_scorer: !new:speechbrain.decoders.scorer.CTCScorer
126
+ eos_index: !ref <eos_index>
127
+ blank_index: !ref <blank_index>
128
+ ctc_fc: !ref <ctc_lin>
129
+
130
+ scorer: !new:speechbrain.decoders.scorer.ScorerBuilder
131
+ full_scorers: [!ref <transformerlm_scorer>, !ref <ctc_scorer>]
132
+ weights:
133
+ transformerlm: !ref <lm_weight>
134
+ ctc: !ref <ctc_weight_decode>
135
+
136
+ decoder: !new:speechbrain.decoders.S2STransformerBeamSearcher
137
+ modules: [!ref <Transformer>, !ref <seq_lin>]
138
  bos_index: !ref <bos_index>
139
  eos_index: !ref <eos_index>
 
140
  min_decode_ratio: !ref <min_decode_ratio>
141
  max_decode_ratio: !ref <max_decode_ratio>
142
  beam_size: !ref <test_beam_size>
 
 
 
143
  temperature: !ref <temperature>
144
  temperature_lm: !ref <temperature_lm>
145
  using_eos_threshold: !ref <using_eos_threshold>
 
147
  length_normalization: !ref <length_normalization>
148
  using_max_attn_shift: !ref <using_max_attn_shift>
149
  max_attn_shift: !ref <max_attn_shift>
150
+ scorer: !ref <scorer>
151
 
152
  modules:
153
  compute_features: !ref <compute_features>