Adel-Moumen
commited on
Update hyperparams.yaml
Browse files- 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
|
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 |
|