Mirco commited on
Commit
7bacef7
1 Parent(s): 3eb53d5

Update hyperparams.yaml

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +1 -0
hyperparams.yaml CHANGED
@@ -60,6 +60,7 @@ CNN: !new:speechbrain.lobes.models.convolution.ConvolutionFrontEnd
60
  kernel_sizes: (3, 3)
61
  strides: (2, 2)
62
  residuals: (False, False)
 
63
 
64
  Transformer: !new:speechbrain.lobes.models.transformer.TransformerASR.TransformerASR # yamllint disable-line rule:line-length
65
  input_size: 5120
 
60
  kernel_sizes: (3, 3)
61
  strides: (2, 2)
62
  residuals: (False, False)
63
+ norm: !name:speechbrain.nnet.normalization.BatchNorm2d
64
 
65
  Transformer: !new:speechbrain.lobes.models.transformer.TransformerASR.TransformerASR # yamllint disable-line rule:line-length
66
  input_size: 5120