Update hyperparams.yaml
Browse files- hyperparams.yaml +12 -2
hyperparams.yaml
CHANGED
|
@@ -20,11 +20,21 @@ emb_size: 128
|
|
| 20 |
dec_neurons: 1024
|
| 21 |
|
| 22 |
# Outputs
|
| 23 |
-
output_neurons:
|
| 24 |
|
| 25 |
# Decoding parameters
|
|
|
|
| 26 |
blank_index: 0
|
| 27 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
|
| 29 |
enc: !new:speechbrain.nnet.containers.Sequential
|
| 30 |
input_shape: [null, null, 1024]
|
|
|
|
| 20 |
dec_neurons: 1024
|
| 21 |
|
| 22 |
# Outputs
|
| 23 |
+
output_neurons: 51 # BPE size, index(blank/eos/bos) = 0
|
| 24 |
|
| 25 |
# Decoding parameters
|
| 26 |
+
# Be sure that the bos and eos index match with the BPEs ones
|
| 27 |
blank_index: 0
|
| 28 |
+
bos_index: 1
|
| 29 |
+
eos_index: 2
|
| 30 |
+
min_decode_ratio: 0.0
|
| 31 |
+
max_decode_ratio: 1.0
|
| 32 |
+
beam_size: 80
|
| 33 |
+
eos_threshold: 1.5
|
| 34 |
+
using_max_attn_shift: True
|
| 35 |
+
max_attn_shift: 140
|
| 36 |
+
ctc_weight_decode: 0.0
|
| 37 |
+
temperature: 1.50
|
| 38 |
|
| 39 |
enc: !new:speechbrain.nnet.containers.Sequential
|
| 40 |
input_shape: [null, null, 1024]
|