speechbrainteam
commited on
Commit
•
f6d184b
1
Parent(s):
9cc14c0
Update hyperparams.yaml
Browse files- hyperparams.yaml +4 -2
hyperparams.yaml
CHANGED
@@ -48,7 +48,7 @@ compute_features: !new:speechbrain.lobes.features.Fbank
|
|
48 |
n_fft: !ref <n_fft>
|
49 |
n_mels: !ref <n_mels>
|
50 |
|
51 |
-
|
52 |
norm_type: global
|
53 |
|
54 |
|
@@ -94,7 +94,7 @@ Tencoder: !new:speechbrain.lobes.models.transformer.TransformerASR.EncoderWrappe
|
|
94 |
encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
95 |
input_shape: [null, null, !ref <n_mels>]
|
96 |
compute_features: !ref <compute_features>
|
97 |
-
normalize: !ref <
|
98 |
cnn: !ref <CNN>
|
99 |
transformer_encoder: !ref <Tencoder>
|
100 |
|
@@ -111,6 +111,7 @@ decoder: !new:speechbrain.decoders.S2STransformerBeamSearch
|
|
111 |
length_normalization: True
|
112 |
|
113 |
modules:
|
|
|
114 |
encoder: !ref <encoder>
|
115 |
decoder: !ref <decoder>
|
116 |
|
@@ -119,5 +120,6 @@ log_softmax: !new:torch.nn.LogSoftmax
|
|
119 |
|
120 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
121 |
loadables:
|
|
|
122 |
asr: !ref <asr_model>
|
123 |
tokenizer: !ref <tokenizer>
|
|
|
48 |
n_fft: !ref <n_fft>
|
49 |
n_mels: !ref <n_mels>
|
50 |
|
51 |
+
normalizer: !new:speechbrain.processing.features.InputNormalization
|
52 |
norm_type: global
|
53 |
|
54 |
|
|
|
94 |
encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
95 |
input_shape: [null, null, !ref <n_mels>]
|
96 |
compute_features: !ref <compute_features>
|
97 |
+
normalize: !ref <normalizer>
|
98 |
cnn: !ref <CNN>
|
99 |
transformer_encoder: !ref <Tencoder>
|
100 |
|
|
|
111 |
length_normalization: True
|
112 |
|
113 |
modules:
|
114 |
+
normalizer: !ref <normalizer>
|
115 |
encoder: !ref <encoder>
|
116 |
decoder: !ref <decoder>
|
117 |
|
|
|
120 |
|
121 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
122 |
loadables:
|
123 |
+
normalizer: !ref <normalizer>
|
124 |
asr: !ref <asr_model>
|
125 |
tokenizer: !ref <tokenizer>
|