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 |
|