poonehmousavi
commited on
Commit
·
49e611d
1
Parent(s):
0847e95
Update hyperparams.yaml
Browse files- 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 |
-
|
84 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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:
|