Adel-Moumen commited on
Commit
0aedced
1 Parent(s): 24338b0

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +14 -5
hyperparams.yaml CHANGED
@@ -99,17 +99,26 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
99
  cnn: !ref <CNN>
100
  transformer_encoder: !ref <Tencoder>
101
 
102
- decoder: !new:speechbrain.decoders.S2STransformerBeamSearch
103
- modules: [!ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
104
- bos_index: !ref <bos_index>
105
  eos_index: !ref <eos_index>
106
  blank_index: !ref <blank_index>
 
 
 
 
 
 
 
 
 
 
 
107
  min_decode_ratio: !ref <min_decode_ratio>
108
  max_decode_ratio: !ref <max_decode_ratio>
109
  beam_size: !ref <test_beam_size>
110
- ctc_weight: !ref <ctc_weight_decode>
111
  using_eos_threshold: False
112
  length_normalization: True
 
113
 
114
  modules:
115
  normalizer: !ref <normalizer>
@@ -123,4 +132,4 @@ pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
123
  loadables:
124
  normalizer: !ref <normalizer>
125
  asr: !ref <asr_model>
126
- tokenizer: !ref <tokenizer>
 
99
  cnn: !ref <CNN>
100
  transformer_encoder: !ref <Tencoder>
101
 
102
+ ctc_scorer: !new:speechbrain.decoders.scorer.CTCScorer
 
 
103
  eos_index: !ref <eos_index>
104
  blank_index: !ref <blank_index>
105
+ ctc_fc: !ref <ctc_lin>
106
+
107
+ scorer: !new:speechbrain.decoders.scorer.ScorerBuilder
108
+ full_scorers: [!ref <ctc_scorer>]
109
+ weights:
110
+ ctc: !ref <ctc_weight_decode>
111
+
112
+ decoder: !new:speechbrain.decoders.S2STransformerBeamSearcher
113
+ modules: [!ref <Transformer>, !ref <seq_lin>]
114
+ bos_index: !ref <bos_index>
115
+ eos_index: !ref <eos_index>
116
  min_decode_ratio: !ref <min_decode_ratio>
117
  max_decode_ratio: !ref <max_decode_ratio>
118
  beam_size: !ref <test_beam_size>
 
119
  using_eos_threshold: False
120
  length_normalization: True
121
+ scorer: !ref <scorer>
122
 
123
  modules:
124
  normalizer: !ref <normalizer>
 
132
  loadables:
133
  normalizer: !ref <normalizer>
134
  asr: !ref <asr_model>
135
+ tokenizer: !ref <tokenizer>