Adel-Moumen
commited on
Commit
•
bfb19f3
1
Parent(s):
1f18745
Update hyperparams.yaml
Browse files- 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 |
-
|
79 |
-
|
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 |
|