Titouan commited on
Commit
0f99f2d
·
verified ·
1 Parent(s): 3d2520d

Revert changes from PR

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +8 -0
hyperparams.yaml CHANGED
@@ -5,6 +5,9 @@
5
  # Feature parameters
6
  n_mels: 80
7
 
 
 
 
8
  # Output parameters
9
  out_n_neurons: 7205
10
 
@@ -48,3 +51,8 @@ pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
48
  mean_var_norm_emb: !ref <mean_var_norm_emb>
49
  classifier: !ref <classifier>
50
  label_encoder: !ref <label_encoder>
 
 
 
 
 
 
5
  # Feature parameters
6
  n_mels: 80
7
 
8
+ # Pretrain folder (HuggingFace)
9
+ pretrained_path: speechbrain/spkrec-ecapa-voxceleb
10
+
11
  # Output parameters
12
  out_n_neurons: 7205
13
 
 
51
  mean_var_norm_emb: !ref <mean_var_norm_emb>
52
  classifier: !ref <classifier>
53
  label_encoder: !ref <label_encoder>
54
+ paths:
55
+ embedding_model: !ref <pretrained_path>/embedding_model.ckpt
56
+ mean_var_norm_emb: !ref <pretrained_path>/mean_var_norm_emb.ckpt
57
+ classifier: !ref <pretrained_path>/classifier.ckpt
58
+ label_encoder: !ref <pretrained_path>/label_encoder.txt