Porjaz commited on
Commit
e305a79
1 Parent(s): 4bad4f3

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. 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>