Adel-Moumen
commited on
Commit
•
c3eaae3
1
Parent(s):
8dbfc21
Update hyperparams.yaml
Browse files- hyperparams.yaml +17 -6
hyperparams.yaml
CHANGED
@@ -119,17 +119,27 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
|
119 |
cnn: !ref <CNN>
|
120 |
transformer_encoder: !ref <Tencoder>
|
121 |
|
122 |
-
|
123 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
124 |
bos_index: !ref <bos_index>
|
125 |
eos_index: !ref <eos_index>
|
126 |
-
blank_index: !ref <blank_index>
|
127 |
min_decode_ratio: !ref <min_decode_ratio>
|
128 |
max_decode_ratio: !ref <max_decode_ratio>
|
129 |
beam_size: !ref <test_beam_size>
|
130 |
-
ctc_weight: !ref <ctc_weight_decode>
|
131 |
-
lm_weight: !ref <lm_weight>
|
132 |
-
lm_modules: !ref <lm_model>
|
133 |
temperature: !ref <temperature>
|
134 |
temperature_lm: !ref <temperature_lm>
|
135 |
using_eos_threshold: !ref <using_eos_threshold>
|
@@ -137,6 +147,7 @@ decoder: !new:speechbrain.decoders.S2STransformerBeamSearch
|
|
137 |
length_normalization: !ref <length_normalization>
|
138 |
using_max_attn_shift: !ref <using_max_attn_shift>
|
139 |
max_attn_shift: !ref <max_attn_shift>
|
|
|
140 |
|
141 |
modules:
|
142 |
compute_features: !ref <compute_features>
|
|
|
119 |
cnn: !ref <CNN>
|
120 |
transformer_encoder: !ref <Tencoder>
|
121 |
|
122 |
+
transformerlm_scorer: !new:speechbrain.decoders.scorer.TransformerLMScorer
|
123 |
+
language_model: !ref <lm_model>
|
124 |
+
|
125 |
+
ctc_scorer: !new:speechbrain.decoders.scorer.CTCScorer
|
126 |
+
eos_index: !ref <eos_index>
|
127 |
+
blank_index: !ref <blank_index>
|
128 |
+
ctc_fc: !ref <ctc_lin>
|
129 |
+
|
130 |
+
scorer: !new:speechbrain.decoders.scorer.ScorerBuilder
|
131 |
+
full_scorers: [!ref <transformerlm_scorer>, !ref <ctc_scorer>]
|
132 |
+
weights:
|
133 |
+
transformerlm: !ref <lm_weight>
|
134 |
+
ctc: !ref <ctc_weight_decode>
|
135 |
+
|
136 |
+
decoder: !new:speechbrain.decoders.S2STransformerBeamSearcher
|
137 |
+
modules: [!ref <Transformer>, !ref <seq_lin>]
|
138 |
bos_index: !ref <bos_index>
|
139 |
eos_index: !ref <eos_index>
|
|
|
140 |
min_decode_ratio: !ref <min_decode_ratio>
|
141 |
max_decode_ratio: !ref <max_decode_ratio>
|
142 |
beam_size: !ref <test_beam_size>
|
|
|
|
|
|
|
143 |
temperature: !ref <temperature>
|
144 |
temperature_lm: !ref <temperature_lm>
|
145 |
using_eos_threshold: !ref <using_eos_threshold>
|
|
|
147 |
length_normalization: !ref <length_normalization>
|
148 |
using_max_attn_shift: !ref <using_max_attn_shift>
|
149 |
max_attn_shift: !ref <max_attn_shift>
|
150 |
+
scorer: !ref <scorer>
|
151 |
|
152 |
modules:
|
153 |
compute_features: !ref <compute_features>
|