poonehmousavi commited on
Commit
49e611d
1 Parent(s): 0847e95

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +17 -2
hyperparams.yaml CHANGED
@@ -80,8 +80,23 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
80
  modules:
81
  encoder: !ref <encoder>
82
 
83
- decoding_function: !name:speechbrain.decoders.ctc_greedy_decode
84
- blank_id: !ref <blank_index>
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
85
 
86
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
87
  loadables:
 
80
  modules:
81
  encoder: !ref <encoder>
82
 
83
+ # Decoding parameters
84
+ decoding_type : beam #"beam" or "greedy"
85
+
86
+ test_beam_search:
87
+ blank_index: !ref <blank_index>
88
+ beam_size: 100
89
+ beam_prune_logp: -12.0
90
+ token_prune_min_logp: -1.2
91
+ prune_history: True
92
+ topk: 1
93
+ alpha: 1.0
94
+ beta: 0.5
95
+ # can be downloaded from here https://www.openslr.org/11/ or trained with kenLM
96
+ # It can either be a .bin or .arpa ; note: .arpa is much slower at loading
97
+ # If you don't want to use an LM, comment it out or set it to null
98
+ kenlm_model_path: en_5gram.arpa
99
+
100
 
101
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
102
  loadables: