gorinars commited on
Commit
86584a1
1 Parent(s): e54489b

first commit

Browse files
Files changed (4) hide show
  1. CKPT.yaml +4 -0
  2. config.json +3 -0
  3. embedding_model.ckpt +3 -0
  4. hyperparams.yaml +52 -0
CKPT.yaml ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ # yamllint disable
2
+ ErrorRate: 0.3550308346748352
3
+ end-of-epoch: true
4
+ unixtime: 1674316424.7192397
config.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "speechbrain_interface": "SpeakerRecognition"
3
+ }
embedding_model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1597dad75036888d38e27f46c3ae4724c742758e604b859a64979b248ad16867
3
+ size 83310835
hyperparams.yaml ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # ############################################################################
2
+ # Model: ECAPA big for Speaker verification
3
+ # ############################################################################
4
+
5
+ # Feature parameters
6
+ n_mels: 80
7
+
8
+ # Pretrain folder (HuggingFace)
9
+ pretrained_path: gorinars/sb-ecapa-vggsound
10
+
11
+ # Output parameters
12
+ out_n_neurons: 308
13
+
14
+ # Model params
15
+ compute_features: !new:speechbrain.lobes.features.Fbank
16
+ n_mels: !ref <n_mels>
17
+
18
+ mean_var_norm: !new:speechbrain.processing.features.InputNormalization
19
+ norm_type: sentence
20
+ std_norm: False
21
+
22
+ embedding_model: !new:speechbrain.lobes.models.ECAPA_TDNN.ECAPA_TDNN
23
+ input_size: !ref <n_mels>
24
+ channels: [1024, 1024, 1024, 1024, 3072]
25
+ kernel_sizes: [5, 3, 3, 3, 1]
26
+ dilations: [1, 2, 3, 4, 1]
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
+ modules:
35
+ compute_features: !ref <compute_features>
36
+ mean_var_norm: !ref <mean_var_norm>
37
+ embedding_model: !ref <embedding_model>
38
+ classifier: !ref <classifier>
39
+
40
+ label_encoder: !new:speechbrain.dataio.encoder.CategoricalEncoder
41
+
42
+
43
+ pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
44
+ loadables:
45
+ embedding_model: !ref <embedding_model>
46
+ #classifier: !ref <classifier>
47
+ #label_encoder: !ref <label_encoder>
48
+ paths:
49
+ embedding_model: !ref <pretrained_path>/embedding_model.ckpt
50
+ #classifier: !ref <pretrained_path>/classifier.ckpt
51
+ #label_encoder: !ref <pretrained_path>/label_encoder.txt
52
+