Adel-Moumen commited on
Commit
2857446
1 Parent(s): c00ab79

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +23 -6
hyperparams.yaml CHANGED
@@ -76,19 +76,36 @@ seq_lin: !new:speechbrain.nnet.linear.Linear
76
  input_size: !ref <d_model>
77
  n_neurons: !ref <output_neurons>
78
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
79
  decoder: !new:speechbrain.decoders.S2STransformerBeamSearch
80
- modules: [!ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
81
  bos_index: !ref <bos_index>
82
  eos_index: !ref <eos_index>
83
- blank_index: !ref <blank_index>
84
  min_decode_ratio: !ref <min_decode_ratio>
85
  max_decode_ratio: !ref <max_decode_ratio>
86
  beam_size: !ref <test_beam_size>
87
- ctc_weight: !ref <ctc_weight_decode>
88
- lm_weight: !ref <lm_weight>
89
- lm_modules: !ref <lm_model>
90
  temperature: 1.15
91
- temperature_lm: 1.15
92
  using_eos_threshold: False
93
  length_normalization: True
94
 
 
76
  input_size: !ref <d_model>
77
  n_neurons: !ref <output_neurons>
78
 
79
+ # Scorer
80
+ ctc_scorer: !new:speechbrain.decoders.scorer.CTCScorer
81
+ eos_index: !ref <eos_index>
82
+ blank_index: !ref <blank_index>
83
+ ctc_fc: !ref <ctc_lin>
84
+
85
+ coverage_scorer: !new:speechbrain.decoders.scorer.CoverageScorer
86
+ vocab_size: !ref <output_neurons>
87
+
88
+ rnnlm_scorer: !new:speechbrain.decoders.scorer.TransformerLMScorer
89
+ language_model: !ref <lm_model>
90
+ temperature: 1.15
91
+
92
+ scorer: !new:speechbrain.decoders.scorer.ScorerBuilder
93
+ full_scorers: [!ref <rnnlm_scorer>,
94
+ !ref <ctc_scorer>,
95
+ !ref <coverage_scorer>]
96
+ weights:
97
+ rnnlm: !ref <lm_weight>
98
+ ctc: !ref <ctc_weight_decode>
99
+ coverage: !ref <coverage_penalty>
100
+
101
  decoder: !new:speechbrain.decoders.S2STransformerBeamSearch
102
+ modules: [!ref <Transformer>, !ref <seq_lin>]
103
  bos_index: !ref <bos_index>
104
  eos_index: !ref <eos_index>
 
105
  min_decode_ratio: !ref <min_decode_ratio>
106
  max_decode_ratio: !ref <max_decode_ratio>
107
  beam_size: !ref <test_beam_size>
 
 
 
108
  temperature: 1.15
 
109
  using_eos_threshold: False
110
  length_normalization: True
111