panchajanya-ai commited on
Commit
00c7225
·
verified ·
1 Parent(s): d060f90

uploading model files

Browse files
classifier.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5babf91bd545d52fe5720e43c5f12bc6ee18a615b2d1cd993b81a9c1a00d166
3
+ size 5163
embedding_model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d3e54c9c361becc6dd2427226e81f59dbdb775338848e31568b4458170d4259
3
+ size 83313724
hyperparams.yaml ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # ############################################################################
2
+ # Model: ECAPA-TDNN for Language Identification
3
+ # ############################################################################
4
+
5
+ # Pretrain folder
6
+ pretrained_path: /model_repository/lid_model/1/v8
7
+
8
+ # Feature parameters
9
+ n_mels: 80
10
+
11
+ # Output parameters
12
+ out_n_neurons: 5 # Possible languages in the dataset
13
+
14
+
15
+ # Model params
16
+ compute_features: !new:speechbrain.lobes.features.Fbank
17
+ n_mels: !ref <n_mels>
18
+
19
+ mean_var_norm: !new:speechbrain.processing.features.InputNormalization
20
+ norm_type: sentence
21
+ std_norm: False
22
+
23
+ embedding_model: !new:speechbrain.lobes.models.ECAPA_TDNN.ECAPA_TDNN
24
+ input_size: !ref <n_mels>
25
+ channels: [1024, 1024, 1024, 1024, 3072]
26
+ kernel_sizes: [5, 3, 3, 3, 1]
27
+ dilations: [1, 2, 3, 4, 1]
28
+ attention_channels: 128
29
+ lin_neurons: 192
30
+
31
+ classifier: !new:speechbrain.lobes.models.ECAPA_TDNN.Classifier
32
+ input_size: 192
33
+ out_neurons: !ref <out_n_neurons>
34
+
35
+ modules:
36
+ compute_features: !ref <compute_features>
37
+ mean_var_norm: !ref <mean_var_norm>
38
+ embedding_model: !ref <embedding_model>
39
+ classifier: !ref <classifier>
40
+
41
+ label_encoder: !new:speechbrain.dataio.encoder.CategoricalEncoder
42
+
43
+
44
+ pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
45
+ loadables:
46
+ embedding_model: !ref <embedding_model>
47
+ classifier: !ref <classifier>
48
+ label_encoder: !ref <label_encoder>
49
+ paths:
50
+ embedding_model: !ref <pretrained_path>/embedding_model.ckpt
51
+ classifier: !ref <pretrained_path>/classifier.ckpt
52
+ label_encoder: !ref <pretrained_path>/language_encoder.txt
language_encoder.txt ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ 'marathi' => 0
2
+ 'hindi' => 1
3
+ 'english' => 2
4
+ 'tamil' => 3
5
+ 'telugu' => 4
6
+ ================
7
+ 'starting_index' => 0