Update hyperparams.yaml

hyperparams.yaml

@@ -30,32 +30,32 @@ bos_index: 1
 eos_index: 2
 
 enc: !new:speechbrain.nnet.containers.Sequential
-  input_shape: [null, null, !ref <wav2vec_output_dim>]
-  linear1: !name:speechbrain.nnet.linear.Linear
-    n_neurons: !ref <dnn_neurons>
-    bias: True
-  bn1: !name:speechbrain.nnet.normalization.BatchNorm1d
-  activation: !new:torch.nn.LeakyReLU
-  drop: !new:torch.nn.Dropout
-    p: !ref <dropout>
-  linear2: !name:speechbrain.nnet.linear.Linear
-    n_neurons: !ref <dnn_neurons>
-    bias: True
-  bn2: !name:speechbrain.nnet.normalization.BatchNorm1d
-  activation2: !new:torch.nn.LeakyReLU
-  drop2: !new:torch.nn.Dropout
-    p: !ref <dropout>
-  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>
-  output_norm: True
-  freeze: True
-  save_path: wav2vec2_checkpoint
+    input_shape: [null, null, !ref <wav2vec_output_dim>]
+    linear1: !name:speechbrain.nnet.linear.Linear
+        n_neurons: !ref <dnn_neurons>
+        bias: True
+    bn1: !name:speechbrain.nnet.normalization.BatchNorm1d
+    activation: !new:torch.nn.LeakyReLU
+    drop: !new:torch.nn.Dropout
+        p: !ref <dropout>
+    linear2: !name:speechbrain.nnet.linear.Linear
+        n_neurons: !ref <dnn_neurons>
+        bias: True
+    bn2: !name:speechbrain.nnet.normalization.BatchNorm1d
+    activation2: !new:torch.nn.LeakyReLU
+    drop2: !new:torch.nn.Dropout
+        p: !ref <dropout>
+    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_transformers.wav2vec2.Wav2Vec2
+    source: !ref <wav2vec2_hub>
+    output_norm: True
+    freeze: True
+    save_path: wav2vec2_checkpoint
 
 ctc_lin: !new:speechbrain.nnet.linear.Linear
   input_size: !ref <dnn_neurons>