mirco commited on
Commit
92ad335
1 Parent(s): b8937e0

update to support classification

Browse files
Files changed (2) hide show
  1. hyperparams.yaml +22 -1
  2. label_encoder.txt +0 -0
hyperparams.yaml CHANGED
@@ -5,6 +5,12 @@
5
  # Feature parameters
6
  n_mels: 80
7
 
 
 
 
 
 
 
8
  # Model params
9
  compute_features: !new:speechbrain.lobes.features.Fbank
10
  n_mels: !ref <n_mels>
@@ -21,6 +27,10 @@ embedding_model: !new:speechbrain.lobes.models.ECAPA_TDNN.ECAPA_TDNN
21
  attention_channels: 128
22
  lin_neurons: 192
23
 
 
 
 
 
24
  mean_var_norm_emb: !new:speechbrain.processing.features.InputNormalization
25
  norm_type: global
26
  std_norm: False
@@ -31,7 +41,18 @@ modules:
31
  embedding_model: !ref <embedding_model>
32
  mean_var_norm_emb: !ref <mean_var_norm_emb>
33
 
 
 
 
34
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
35
  loadables:
36
  embedding_model: !ref <embedding_model>
37
- mean_var_norm_emb: !ref <mean_var_norm_emb>
 
 
 
 
 
 
 
 
 
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
+
14
  # Model params
15
  compute_features: !new:speechbrain.lobes.features.Fbank
16
  n_mels: !ref <n_mels>
 
27
  attention_channels: 128
28
  lin_neurons: 192
29
 
30
+ classifier: !new:speechbrain.lobes.models.ECAPA_TDNN.Classifier
31
+ input_size: 192
32
+ out_neurons: !ref <out_n_neurons>
33
+
34
  mean_var_norm_emb: !new:speechbrain.processing.features.InputNormalization
35
  norm_type: global
36
  std_norm: False
 
41
  embedding_model: !ref <embedding_model>
42
  mean_var_norm_emb: !ref <mean_var_norm_emb>
43
 
44
+ label_encoder: !new:speechbrain.dataio.encoder.CategoricalEncoder
45
+
46
+
47
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
48
  loadables:
49
  embedding_model: !ref <embedding_model>
50
+ mean_var_norm_emb: !ref <mean_var_norm_emb>
51
+ classifier: !ref <classifier>
52
+ label_encoder: !ref <label_encoder>
53
+ paths:
54
+ embedding_model: !ref <pretrained_path>/embedding_model.ckpt
55
+ mean_var_norm_emb: !ref <pretrained_path>/mean_var_norm_emb.ckpt
56
+ classifier: !ref <pretrained_path>/classifier.ckpt
57
+ label_encoder: !ref <pretrained_path>/label_encoder.txt
58
+
label_encoder.txt ADDED
The diff for this file is too large to render. See raw diff