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>
|