Ubenwa
/

sb-ecapa-dummy-tiny

Model card Files Files and versions Community

gorinars commited on Mar 13, 2023

Commit

b6d03f6

•

1 Parent(s): 238d1fe

Update hyperparams.yaml

Files changed (1) hide show

hyperparams.yaml +5 -5

hyperparams.yaml CHANGED Viewed

@@ -21,14 +21,14 @@ mean_var_norm: !new:speechbrain.processing.features.InputNormalization
 embedding_model: !new:speechbrain.lobes.models.ECAPA_TDNN.ECAPA_TDNN
     input_size: !ref <n_mels>
-    channels: [4, 4, 4, 4, 12]
     kernel_sizes: [5, 3, 3, 3, 1]
     dilations: [1, 2, 3, 4, 1]
-    attention_channels: 4
-    lin_neurons: 8
 classifier: !new:speechbrain.lobes.models.ECAPA_TDNN.Classifier
-    input_size: 8
     out_neurons: !ref <out_n_neurons>
 modules:

 embedding_model: !new:speechbrain.lobes.models.ECAPA_TDNN.ECAPA_TDNN
     input_size: !ref <n_mels>
+    channels: [8, 8, 8, 8, 24]
     kernel_sizes: [5, 3, 3, 3, 1]
     dilations: [1, 2, 3, 4, 1]
+    attention_channels: 2
+    lin_neurons: 2
 classifier: !new:speechbrain.lobes.models.ECAPA_TDNN.Classifier
+    input_size: 2
     out_neurons: !ref <out_n_neurons>
 modules: