Revert changes from PR
Browse files- 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
|