Hotfixe for Samuele' PR on transformer librispeech.
Browse files- hyperparams.yaml +1 -1
hyperparams.yaml
CHANGED
@@ -52,7 +52,7 @@ CNN: !new:speechbrain.lobes.models.convolution.ConvolutionFrontEnd
|
|
52 |
kernel_sizes: (3, 3, 1)
|
53 |
strides: (2, 2, 1)
|
54 |
residuals: (False, False, False)
|
55 |
-
norm: BatchNorm2d
|
56 |
|
57 |
Transformer: !new:speechbrain.lobes.models.transformer.TransformerASR.TransformerASR
|
58 |
input_size: 10240
|
|
|
52 |
kernel_sizes: (3, 3, 1)
|
53 |
strides: (2, 2, 1)
|
54 |
residuals: (False, False, False)
|
55 |
+
norm: !name:speechbrain.nnet.normalization.BatchNorm2d
|
56 |
|
57 |
Transformer: !new:speechbrain.lobes.models.transformer.TransformerASR.TransformerASR
|
58 |
input_size: 10240
|