Update hyperparams.yaml
Browse files- hyperparams.yaml +12 -3
hyperparams.yaml
CHANGED
@@ -76,19 +76,28 @@ 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 <beam_size>
|
87 |
-
ctc_weight: !ref <ctc_weight_decode>
|
88 |
temperature: 1.15
|
89 |
temperature_lm: 1.15
|
90 |
using_eos_threshold: False
|
91 |
length_normalization: True
|
|
|
92 |
|
93 |
log_softmax: !new:torch.nn.LogSoftmax
|
94 |
dim: -1
|
|
|
76 |
input_size: !ref <d_model>
|
77 |
n_neurons: !ref <output_neurons>
|
78 |
|
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 |
+
scorer: !new:speechbrain.decoders.scorer.ScorerBuilder
|
85 |
+
full_scorers: [!ref <ctc_scorer>]
|
86 |
+
weights:
|
87 |
+
ctc: !ref <ctc_weight_decode>
|
88 |
+
|
89 |
decoder: !new:speechbrain.decoders.S2STransformerBeamSearch
|
90 |
+
modules: [!ref <Transformer>, !ref <seq_lin>]
|
91 |
bos_index: !ref <bos_index>
|
92 |
eos_index: !ref <eos_index>
|
|
|
93 |
min_decode_ratio: !ref <min_decode_ratio>
|
94 |
max_decode_ratio: !ref <max_decode_ratio>
|
95 |
beam_size: !ref <beam_size>
|
|
|
96 |
temperature: 1.15
|
97 |
temperature_lm: 1.15
|
98 |
using_eos_threshold: False
|
99 |
length_normalization: True
|
100 |
+
scorer: !ref <scorer>
|
101 |
|
102 |
log_softmax: !new:torch.nn.LogSoftmax
|
103 |
dim: -1
|