speechbrainteam
commited on
Commit
·
457789e
1
Parent(s):
3e1cd7a
Update hyperparams.yaml
Browse files- hyperparams.yaml +4 -4
hyperparams.yaml
CHANGED
@@ -92,7 +92,7 @@ decoder: !new:speechbrain.decoders.S2STransformerBeamSearch
|
|
92 |
log_softmax: !new:torch.nn.LogSoftmax
|
93 |
dim: -1
|
94 |
|
95 |
-
|
96 |
norm_type: global
|
97 |
|
98 |
compute_features: !new:speechbrain.lobes.features.Fbank
|
@@ -123,7 +123,7 @@ Tencoder: !new:speechbrain.lobes.models.transformer.TransformerASR.EncoderWrappe
|
|
123 |
encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
124 |
input_shape: [null, null, !ref <n_mels>]
|
125 |
compute_features: !ref <compute_features>
|
126 |
-
normalize: !ref <
|
127 |
cnn: !ref <CNN>
|
128 |
transformer_encoder: !ref <Tencoder>
|
129 |
|
@@ -136,7 +136,7 @@ modules:
|
|
136 |
pre_transformer: !ref <CNN>
|
137 |
transformer: !ref <Transformer>
|
138 |
asr_model: !ref <asr_model>
|
139 |
-
|
140 |
lm_model: !ref <lm_model>
|
141 |
encoder: !ref <encoder>
|
142 |
decoder: !ref <decoder>
|
@@ -145,7 +145,7 @@ modules:
|
|
145 |
# are declared in the yaml.
|
146 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
147 |
loadables:
|
148 |
-
|
149 |
asr: !ref <asr_model>
|
150 |
lm: !ref <lm_model>
|
151 |
tokenizer: !ref <tokenizer>
|
|
|
92 |
log_softmax: !new:torch.nn.LogSoftmax
|
93 |
dim: -1
|
94 |
|
95 |
+
normalizer: !new:speechbrain.processing.features.InputNormalization
|
96 |
norm_type: global
|
97 |
|
98 |
compute_features: !new:speechbrain.lobes.features.Fbank
|
|
|
123 |
encoder: !new:speechbrain.nnet.containers.LengthsCapableSequential
|
124 |
input_shape: [null, null, !ref <n_mels>]
|
125 |
compute_features: !ref <compute_features>
|
126 |
+
normalize: !ref <normalizer>
|
127 |
cnn: !ref <CNN>
|
128 |
transformer_encoder: !ref <Tencoder>
|
129 |
|
|
|
136 |
pre_transformer: !ref <CNN>
|
137 |
transformer: !ref <Transformer>
|
138 |
asr_model: !ref <asr_model>
|
139 |
+
normalizer: !ref <normalizer>
|
140 |
lm_model: !ref <lm_model>
|
141 |
encoder: !ref <encoder>
|
142 |
decoder: !ref <decoder>
|
|
|
145 |
# are declared in the yaml.
|
146 |
pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
|
147 |
loadables:
|
148 |
+
normalizer: !ref <normalizer>
|
149 |
asr: !ref <asr_model>
|
150 |
lm: !ref <lm_model>
|
151 |
tokenizer: !ref <tokenizer>
|