Titouan commited on
Commit
51ae62d
1 Parent(s): 3609bd5

Pre-patch for Samuele's update on Transformer for Librispeech

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +1 -0
hyperparams.yaml CHANGED
@@ -52,6 +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
 
56
  Transformer: !new:speechbrain.lobes.models.transformer.TransformerASR.TransformerASR
57
  input_size: 10240
52
  kernel_sizes: (3, 3, 1)
53
  strides: (2, 2, 1)
54
  residuals: (False, False, False)
55
+ norm: speechbrain.nnet.normalization.BatchNorm2d
56
 
57
  Transformer: !new:speechbrain.lobes.models.transformer.TransformerASR.TransformerASR
58
  input_size: 10240