gorinars commited on
Commit
53f08d4
1 Parent(s): c620532

first commit

Browse files
Files changed (3) hide show
  1. .gitattributes +1 -0
  2. embedding_model.ckpt +3 -0
  3. hyperparams.yaml +43 -0
.gitattributes CHANGED
@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
32
  *.zip filter=lfs diff=lfs merge=lfs -text
33
  *.zst filter=lfs diff=lfs merge=lfs -text
34
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
32
  *.zip filter=lfs diff=lfs merge=lfs -text
33
  *.zst filter=lfs diff=lfs merge=lfs -text
34
  *tfevents* filter=lfs diff=lfs merge=lfs -text
35
+ embedding_model.ckpt filter=lfs diff=lfs merge=lfs -text
embedding_model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb4388583b192cbca56198b1e5d2bd78336210ee7c4b6de2019d1d09b7bfeb00
3
+ size 83310835
hyperparams.yaml ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # ############################################################################
2
+ # Model: ECAPA VoxCeleb finetuned on CryCeleb to verify babies
3
+ # ############################################################################
4
+
5
+ # Feature parameters
6
+ n_mels: 80
7
+
8
+ # Pretrain folder (HuggingFace)
9
+ pretrained_path: Ubenwa/ecapa-voxceleb-ft-cryceleb
10
+
11
+ # Model params
12
+ compute_features: !new:speechbrain.lobes.features.Fbank
13
+ n_mels: !ref <n_mels>
14
+
15
+ mean_var_norm: !new:speechbrain.processing.features.InputNormalization
16
+ norm_type: sentence
17
+ std_norm: False
18
+
19
+ embedding_model: !new:speechbrain.lobes.models.ECAPA_TDNN.ECAPA_TDNN
20
+ input_size: !ref <n_mels>
21
+ channels: [1024, 1024, 1024, 1024, 3072]
22
+ kernel_sizes: [5, 3, 3, 3, 1]
23
+ dilations: [1, 2, 3, 4, 1]
24
+ attention_channels: 128
25
+ lin_neurons: 192
26
+
27
+ classifier: !new:speechbrain.lobes.models.ECAPA_TDNN.Classifier
28
+ input_size: 192
29
+ out_neurons: !ref <out_n_neurons>
30
+
31
+ modules:
32
+ compute_features: !ref <compute_features>
33
+ mean_var_norm: !ref <mean_var_norm>
34
+ embedding_model: !ref <embedding_model>
35
+
36
+ label_encoder: !new:speechbrain.dataio.encoder.CategoricalEncoder
37
+
38
+
39
+ pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
40
+ loadables:
41
+ embedding_model: !ref <embedding_model>
42
+ paths:
43
+ embedding_model: !ref <pretrained_path>/embedding_model.ckpt