Adel-Moumen
commited on
Commit
•
1edc1a1
1
Parent(s):
e7f9f8b
Update hyperparams.yaml
Browse files- hyperparams.yaml +13 -4
hyperparams.yaml
CHANGED
@@ -86,17 +86,26 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
|
86 |
wav2vec2: !ref <wav2vec2>
|
87 |
transformer_encoder: !ref <Tencoder>
|
88 |
|
89 |
-
|
90 |
-
modules: [!ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
|
91 |
-
bos_index: !ref <bos_index>
|
92 |
eos_index: !ref <eos_index>
|
93 |
blank_index: !ref <blank_index>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
94 |
min_decode_ratio: !ref <min_decode_ratio>
|
95 |
max_decode_ratio: !ref <max_decode_ratio>
|
96 |
beam_size: !ref <test_beam_size>
|
97 |
-
ctc_weight: !ref <ctc_weight_decode>
|
98 |
using_eos_threshold: False
|
99 |
length_normalization: True
|
|
|
100 |
|
101 |
modules:
|
102 |
encoder: !ref <encoder>
|
|
|
86 |
wav2vec2: !ref <wav2vec2>
|
87 |
transformer_encoder: !ref <Tencoder>
|
88 |
|
89 |
+
ctc_scorer: !new:speechbrain.decoders.scorer.CTCScorer
|
|
|
|
|
90 |
eos_index: !ref <eos_index>
|
91 |
blank_index: !ref <blank_index>
|
92 |
+
ctc_fc: !ref <ctc_lin>
|
93 |
+
|
94 |
+
scorer: !new:speechbrain.decoders.scorer.ScorerBuilder
|
95 |
+
full_scorers: [!ref <ctc_scorer>]
|
96 |
+
weights:
|
97 |
+
ctc: !ref <ctc_weight_decode>
|
98 |
+
|
99 |
+
decoder: !new:speechbrain.decoders.S2STransformerBeamSearcher
|
100 |
+
modules: [!ref <Transformer>, !ref <seq_lin>]
|
101 |
+
bos_index: !ref <bos_index>
|
102 |
+
eos_index: !ref <eos_index>
|
103 |
min_decode_ratio: !ref <min_decode_ratio>
|
104 |
max_decode_ratio: !ref <max_decode_ratio>
|
105 |
beam_size: !ref <test_beam_size>
|
|
|
106 |
using_eos_threshold: False
|
107 |
length_normalization: True
|
108 |
+
scorer: !ref <scorer>
|
109 |
|
110 |
modules:
|
111 |
encoder: !ref <encoder>
|