Titouan commited on
Commit
7c41d58
1 Parent(s): dd5e4d4

Hotfixe for Samuele' PR on transformer librispeech.

Browse files
Files changed (1) hide show
  1. 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