Update hyperparams.yaml

hyperparams.yaml

@@ -35,11 +35,31 @@ max_attn_shift: 140
 ctc_weight_decode: 0.0
 temperature: 1.50
 
-enc: !new:speechbrain.lobes.models.VanillaNN.VanillaNN
+# enc: !new:speechbrain.lobes.models.VanillaNN.VanillaNN
+#     input_shape: [null, null, 1024]
+#     activation: !ref <activation>
+#     dnn_blocks: !ref <dnn_layers>
+#     dnn_neurons: !ref <dnn_neurons>
+
+enc: !new:speechbrain.nnet.containers.Sequential
     input_shape: [null, null, 1024]
-    activation: !ref <activation>
-    dnn_blocks: !ref <dnn_layers>
-    dnn_neurons: !ref <dnn_neurons>
+    linear1: !name:speechbrain.nnet.linear.Linear
+        n_neurons: !ref <dnn_neurons>
+        bias: True
+    bn1: !name:speechbrain.nnet.normalization.BatchNorm1d
+    activation: !new:torch.nn.LeakyReLU
+
+    linear2: !name:speechbrain.nnet.linear.Linear
+        n_neurons: !ref <dnn_neurons>
+        bias: True
+    bn2: !name:speechbrain.nnet.normalization.BatchNorm1d
+    activation2: !new:torch.nn.LeakyReLU
+
+    linear3: !name:speechbrain.nnet.linear.Linear
+        n_neurons: !ref <dnn_neurons>
+        bias: True
+    bn3: !name:speechbrain.nnet.normalization.BatchNorm1d
+    activation3: !new:torch.nn.LeakyReLU
 
 wav2vec2: !new:speechbrain.lobes.models.huggingface_wav2vec.HuggingFaceWav2Vec2
     source: !ref <wav2vec2_hub>