mirco commited on
Commit
7781186
1 Parent(s): b63e7c7

support label encoder pre-train

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +13 -1
hyperparams.yaml CHANGED
@@ -5,8 +5,10 @@
5
  # Feature parameters
6
  n_mels: 24
7
 
 
 
 
8
  # Output parameters
9
- lab_encoder_file: /home/mirco/spkrec-xvect-voxceleb/label_encoder.txt
10
  out_n_neurons: 7205
11
 
12
 
@@ -44,9 +46,19 @@ modules:
44
  embedding_model: !ref <embedding_model>
45
  mean_var_norm_emb: !ref <mean_var_norm_emb>
46
  classifier: !ref <classifier>
 
 
 
47
 
48
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
49
  loadables:
50
  embedding_model: !ref <embedding_model>
51
  mean_var_norm_emb: !ref <mean_var_norm_emb>
52
  classifier: !ref <classifier>
 
 
 
 
 
 
 
 
5
  # Feature parameters
6
  n_mels: 24
7
 
8
+ # Pretrain folder (HuggingFace)
9
+ pretrained_path: speechbrain/spkrec-xvect-voxceleb
10
+
11
  # Output parameters
 
12
  out_n_neurons: 7205
13
 
14
 
 
46
  embedding_model: !ref <embedding_model>
47
  mean_var_norm_emb: !ref <mean_var_norm_emb>
48
  classifier: !ref <classifier>
49
+
50
+ label_encoder: !new:speechbrain.dataio.encoder.CategoricalEncoder
51
+
52
 
53
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
54
  loadables:
55
  embedding_model: !ref <embedding_model>
56
  mean_var_norm_emb: !ref <mean_var_norm_emb>
57
  classifier: !ref <classifier>
58
+ label_encoder: !ref <label_encoder>
59
+ paths:
60
+ embedding_model: !ref <pretrained_path>/embedding_model.ckpt
61
+ mean_var_norm_emb: !ref <pretrained_path>/mean_var_norm_emb.ckpt
62
+ classifier: !ref <pretrained_path>/classifier.ckpt
63
+ label_encoder: !ref <pretrained_path>/label_encoder.txt
64
+