Update hyperparams.yaml
Browse files- hyperparams.yaml +14 -5
hyperparams.yaml
CHANGED
|
@@ -99,17 +99,26 @@ encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
|
| 99 |
cnn: !ref <CNN>
|
| 100 |
transformer_encoder: !ref <Tencoder>
|
| 101 |
|
| 102 |
-
|
| 103 |
-
modules: [!ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
|
| 104 |
-
bos_index: !ref <bos_index>
|
| 105 |
eos_index: !ref <eos_index>
|
| 106 |
blank_index: !ref <blank_index>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 107 |
min_decode_ratio: !ref <min_decode_ratio>
|
| 108 |
max_decode_ratio: !ref <max_decode_ratio>
|
| 109 |
beam_size: !ref <test_beam_size>
|
| 110 |
-
ctc_weight: !ref <ctc_weight_decode>
|
| 111 |
using_eos_threshold: False
|
| 112 |
length_normalization: True
|
|
|
|
| 113 |
|
| 114 |
modules:
|
| 115 |
normalizer: !ref <normalizer>
|
|
@@ -123,4 +132,4 @@ pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
|
| 123 |
loadables:
|
| 124 |
normalizer: !ref <normalizer>
|
| 125 |
asr: !ref <asr_model>
|
| 126 |
-
tokenizer: !ref <tokenizer>
|
|
|
|
| 99 |
cnn: !ref <CNN>
|
| 100 |
transformer_encoder: !ref <Tencoder>
|
| 101 |
|
| 102 |
+
ctc_scorer: !new:speechbrain.decoders.scorer.CTCScorer
|
|
|
|
|
|
|
| 103 |
eos_index: !ref <eos_index>
|
| 104 |
blank_index: !ref <blank_index>
|
| 105 |
+
ctc_fc: !ref <ctc_lin>
|
| 106 |
+
|
| 107 |
+
scorer: !new:speechbrain.decoders.scorer.ScorerBuilder
|
| 108 |
+
full_scorers: [!ref <ctc_scorer>]
|
| 109 |
+
weights:
|
| 110 |
+
ctc: !ref <ctc_weight_decode>
|
| 111 |
+
|
| 112 |
+
decoder: !new:speechbrain.decoders.S2STransformerBeamSearcher
|
| 113 |
+
modules: [!ref <Transformer>, !ref <seq_lin>]
|
| 114 |
+
bos_index: !ref <bos_index>
|
| 115 |
+
eos_index: !ref <eos_index>
|
| 116 |
min_decode_ratio: !ref <min_decode_ratio>
|
| 117 |
max_decode_ratio: !ref <max_decode_ratio>
|
| 118 |
beam_size: !ref <test_beam_size>
|
|
|
|
| 119 |
using_eos_threshold: False
|
| 120 |
length_normalization: True
|
| 121 |
+
scorer: !ref <scorer>
|
| 122 |
|
| 123 |
modules:
|
| 124 |
normalizer: !ref <normalizer>
|
|
|
|
| 132 |
loadables:
|
| 133 |
normalizer: !ref <normalizer>
|
| 134 |
asr: !ref <asr_model>
|
| 135 |
+
tokenizer: !ref <tokenizer>
|