Update hyperparams.yaml
Browse files- hyperparams.yaml +8 -0
hyperparams.yaml
CHANGED
@@ -94,6 +94,14 @@ model: !new:torch.nn.ModuleList
|
|
94 |
- [!ref <encoder_w2v2>, !ref <embedding>, !ref <decoder>, !ref <ctc_lin>, !ref <seq_lin>]
|
95 |
|
96 |
############################## Decoding & optimiser ############################
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
97 |
test_search: !new:speechbrain.decoders.S2SRNNBeamSearcher
|
98 |
embedding: !ref <embedding>
|
99 |
decoder: !ref <decoder>
|
|
|
94 |
- [!ref <encoder_w2v2>, !ref <embedding>, !ref <decoder>, !ref <ctc_lin>, !ref <seq_lin>]
|
95 |
|
96 |
############################## Decoding & optimiser ############################
|
97 |
+
coverage_scorer: !new:speechbrain.decoders.scorer.CoverageScorer
|
98 |
+
vocab_size: !ref <output_neurons>
|
99 |
+
|
100 |
+
scorer: !new:speechbrain.decoders.scorer.ScorerBuilder
|
101 |
+
full_scorers: [!ref <coverage_scorer>]
|
102 |
+
weights:
|
103 |
+
coverage: !ref <coverage_penalty>
|
104 |
+
|
105 |
test_search: !new:speechbrain.decoders.S2SRNNBeamSearcher
|
106 |
embedding: !ref <embedding>
|
107 |
decoder: !ref <decoder>
|