support label encoder pre-train

hyperparams.yaml

@@ -5,8 +5,10 @@
 # Feature parameters
 n_mels: 24
 
+# Pretrain folder (HuggingFace)
+pretrained_path: speechbrain/spkrec-xvect-voxceleb
+
 # Output parameters
-lab_encoder_file: /home/mirco/spkrec-xvect-voxceleb/label_encoder.txt
 out_n_neurons: 7205
 
 
@@ -44,9 +46,19 @@ modules:
     embedding_model: !ref <embedding_model>
     mean_var_norm_emb: !ref <mean_var_norm_emb>
     classifier: !ref <classifier>
+
+label_encoder: !new:speechbrain.dataio.encoder.CategoricalEncoder
+
         
 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>
+        label_encoder: !ref <label_encoder>
+    paths:
+        embedding_model: !ref <pretrained_path>/embedding_model.ckpt
+        mean_var_norm_emb: !ref <pretrained_path>/mean_var_norm_emb.ckpt
+        classifier: !ref <pretrained_path>/classifier.ckpt
+        label_encoder: !ref <pretrained_path>/label_encoder.txt
+