Adel-Moumen commited on
Commit
bfb19f3
1 Parent(s): 1f18745

Update hyperparams.yaml

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