Adel-Moumen
commited on
Commit
•
0aedced
1
Parent(s):
24338b0
Update hyperparams.yaml
Browse files- hyperparams.yaml +14 -5
hyperparams.yaml
CHANGED
@@ -99,17 +99,26 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
|
99 |
cnn: !ref <CNN>
|
100 |
transformer_encoder: !ref <Tencoder>
|
101 |
|
102 |
-
|
103 |
-
modules: [!ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
|
104 |
-
bos_index: !ref <bos_index>
|
105 |
eos_index: !ref <eos_index>
|
106 |
blank_index: !ref <blank_index>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
107 |
min_decode_ratio: !ref <min_decode_ratio>
|
108 |
max_decode_ratio: !ref <max_decode_ratio>
|
109 |
beam_size: !ref <test_beam_size>
|
110 |
-
ctc_weight: !ref <ctc_weight_decode>
|
111 |
using_eos_threshold: False
|
112 |
length_normalization: True
|
|
|
113 |
|
114 |
modules:
|
115 |
normalizer: !ref <normalizer>
|
@@ -123,4 +132,4 @@ pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
|
123 |
loadables:
|
124 |
normalizer: !ref <normalizer>
|
125 |
asr: !ref <asr_model>
|
126 |
-
tokenizer: !ref <tokenizer>
|
|
|
99 |
cnn: !ref <CNN>
|
100 |
transformer_encoder: !ref <Tencoder>
|
101 |
|
102 |
+
ctc_scorer: !new:speechbrain.decoders.scorer.CTCScorer
|
|
|
|
|
103 |
eos_index: !ref <eos_index>
|
104 |
blank_index: !ref <blank_index>
|
105 |
+
ctc_fc: !ref <ctc_lin>
|
106 |
+
|
107 |
+
scorer: !new:speechbrain.decoders.scorer.ScorerBuilder
|
108 |
+
full_scorers: [!ref <ctc_scorer>]
|
109 |
+
weights:
|
110 |
+
ctc: !ref <ctc_weight_decode>
|
111 |
+
|
112 |
+
decoder: !new:speechbrain.decoders.S2STransformerBeamSearcher
|
113 |
+
modules: [!ref <Transformer>, !ref <seq_lin>]
|
114 |
+
bos_index: !ref <bos_index>
|
115 |
+
eos_index: !ref <eos_index>
|
116 |
min_decode_ratio: !ref <min_decode_ratio>
|
117 |
max_decode_ratio: !ref <max_decode_ratio>
|
118 |
beam_size: !ref <test_beam_size>
|
|
|
119 |
using_eos_threshold: False
|
120 |
length_normalization: True
|
121 |
+
scorer: !ref <scorer>
|
122 |
|
123 |
modules:
|
124 |
normalizer: !ref <normalizer>
|
|
|
132 |
loadables:
|
133 |
normalizer: !ref <normalizer>
|
134 |
asr: !ref <asr_model>
|
135 |
+
tokenizer: !ref <tokenizer>
|