update to support classification as well

hyperparams.yaml

@@ -5,6 +5,11 @@
 # Feature parameters
 n_mels: 24
 
+# Output parameters
+lab_encoder_file: /home/mirco/spkrec-xvect-voxceleb/label_encoder.txt
+out_n_neurons: 7205
+
+
 # Model params
 compute_features: !new:speechbrain.lobes.features.Fbank
     n_mels: !ref <n_mels>
@@ -22,6 +27,13 @@ embedding_model: !new:speechbrain.lobes.models.Xvector.Xvector
     tdnn_dilations: [1, 2, 3, 1, 1]
     lin_neurons: 512
 
+classifier: !new:speechbrain.lobes.models.Xvector.Classifier
+    input_shape: [null, null, 512]
+    activation: !name:torch.nn.LeakyReLU
+    lin_blocks: 1
+    lin_neurons: 512
+    out_neurons: !ref <out_n_neurons>
+
 mean_var_norm_emb: !new:speechbrain.processing.features.InputNormalization
     norm_type: global
     std_norm: False
@@ -31,8 +43,10 @@ modules:
     mean_var_norm: !ref <mean_var_norm>
     embedding_model: !ref <embedding_model>
     mean_var_norm_emb: !ref <mean_var_norm_emb>
+    classifier: !ref <classifier>
         
 pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
     loadables:
         embedding_model: !ref <embedding_model>
         mean_var_norm_emb: !ref <mean_var_norm_emb>
+        classifier: !ref <classifier>