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

update to support classification as well

Browse files
Files changed (2) hide show
  1. hyperparams.yaml +14 -0
  2. label_encoder.txt +0 -0
hyperparams.yaml CHANGED
@@ -5,6 +5,11 @@
5
  # Feature parameters
6
  n_mels: 24
7
 
 
 
 
 
 
8
  # Model params
9
  compute_features: !new:speechbrain.lobes.features.Fbank
10
  n_mels: !ref <n_mels>
@@ -22,6 +27,13 @@ embedding_model: !new:speechbrain.lobes.models.Xvector.Xvector
22
  tdnn_dilations: [1, 2, 3, 1, 1]
23
  lin_neurons: 512
24
 
 
 
 
 
 
 
 
25
  mean_var_norm_emb: !new:speechbrain.processing.features.InputNormalization
26
  norm_type: global
27
  std_norm: False
@@ -31,8 +43,10 @@ modules:
31
  mean_var_norm: !ref <mean_var_norm>
32
  embedding_model: !ref <embedding_model>
33
  mean_var_norm_emb: !ref <mean_var_norm_emb>
 
34
 
35
  pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
36
  loadables:
37
  embedding_model: !ref <embedding_model>
38
  mean_var_norm_emb: !ref <mean_var_norm_emb>
 
 
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
+
13
  # Model params
14
  compute_features: !new:speechbrain.lobes.features.Fbank
15
  n_mels: !ref <n_mels>
 
27
  tdnn_dilations: [1, 2, 3, 1, 1]
28
  lin_neurons: 512
29
 
30
+ classifier: !new:speechbrain.lobes.models.Xvector.Classifier
31
+ input_shape: [null, null, 512]
32
+ activation: !name:torch.nn.LeakyReLU
33
+ lin_blocks: 1
34
+ lin_neurons: 512
35
+ out_neurons: !ref <out_n_neurons>
36
+
37
  mean_var_norm_emb: !new:speechbrain.processing.features.InputNormalization
38
  norm_type: global
39
  std_norm: False
 
43
  mean_var_norm: !ref <mean_var_norm>
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>
label_encoder.txt ADDED
The diff for this file is too large to render. See raw diff