Mirco commited on
Commit
1fef18e
1 Parent(s): 459c98a

update model + fix shape

Browse files
Files changed (4) hide show
  1. classifier.ckpt +1 -1
  2. embedding_model.ckpt +1 -1
  3. hyperparams.yaml +1 -1
  4. label_encoder.txt +45 -45
classifier.ckpt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0e5ee21ca315e79dd88e58b3aa00c4e5ca745f74d03526ed2486f0fa59a86fa0
3
  size 35371
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:69ef9f5dfe97676ff8dcfd885ccaae74be7d4bc70115b9d357422f3b82b36137
3
  size 35371
embedding_model.ckpt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a21870cdc658aa94ad54c48bbdf88079d63a699c4cbb4494ea9466896c2bd12f
3
  size 83310835
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf3b5f97e6be1f013aa41c2e7bb7511a6d02a2767ec042a5a3d7f3b99391403d
3
  size 83310835
hyperparams.yaml CHANGED
@@ -23,7 +23,7 @@ mean_var_norm: !new:speechbrain.processing.features.InputNormalization
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, 1, 1]
27
  dilations: [1, 2, 3, 4, 1]
28
  attention_channels: 128
29
  lin_neurons: 192
 
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
label_encoder.txt CHANGED
@@ -1,47 +1,47 @@
1
- 'Arabic' => 0
2
- 'Portuguese' => 1
3
- 'Romansh_Sursilvan' => 2
4
- 'Japanese' => 3
5
- 'Ukranian' => 4
6
- 'German' => 5
7
- 'Chinese_China' => 6
8
- 'Welsh' => 7
9
- 'English' => 8
10
- 'Kabyle' => 9
11
- 'Kyrgyz' => 10
12
- 'Georgian' => 11
13
- 'Persian' => 12
14
- 'French' => 13
15
- 'Interlingua' => 14
16
- 'Swedish' => 15
17
- 'Spanish' => 16
18
- 'Dhivehi' => 17
19
- 'Kinyarwanda' => 18
20
- 'Tatar' => 19
21
- 'Hakha_Chin' => 20
22
- 'Tamil' => 21
23
- 'Greek' => 22
24
- 'Latvian' => 23
25
- 'Russian' => 24
26
- 'Breton' => 25
27
- 'Catalan' => 26
28
- 'Maltese' => 27
29
- 'Slovenian' => 28
30
- 'Indonesian' => 29
31
- 'Dutch' => 30
32
- 'Chinese_Taiwan' => 31
33
- 'Sakha' => 32
34
- 'Polish' => 33
35
- 'Czech' => 34
36
- 'Romanian' => 35
37
- 'Mangolian' => 36
38
- 'Italian' => 37
39
- 'Chinese_Hongkong' => 38
40
- 'Estonian' => 39
41
- 'Basque' => 40
42
- 'Esperanto' => 41
43
- 'Frisian' => 42
44
- 'Turkish' => 43
45
- 'Chuvash' => 44
46
  ================
47
  'starting_index' => 0
 
1
+ 'Basque' => 0
2
+ 'Romansh_Sursilvan' => 1
3
+ 'Sakha' => 2
4
+ 'Georgian' => 3
5
+ 'Greek' => 4
6
+ 'Hakha_Chin' => 5
7
+ 'Ukranian' => 6
8
+ 'Interlingua' => 7
9
+ 'Persian' => 8
10
+ 'Polish' => 9
11
+ 'Dutch' => 10
12
+ 'Chinese_Hongkong' => 11
13
+ 'Japanese' => 12
14
+ 'Portuguese' => 13
15
+ 'Italian' => 14
16
+ 'Catalan' => 15
17
+ 'Chuvash' => 16
18
+ 'Swedish' => 17
19
+ 'Spanish' => 18
20
+ 'Slovenian' => 19
21
+ 'Tamil' => 20
22
+ 'Breton' => 21
23
+ 'Russian' => 22
24
+ 'Czech' => 23
25
+ 'English' => 24
26
+ 'French' => 25
27
+ 'Tatar' => 26
28
+ 'Welsh' => 27
29
+ 'Kyrgyz' => 28
30
+ 'Esperanto' => 29
31
+ 'Kinyarwanda' => 30
32
+ 'Dhivehi' => 31
33
+ 'Turkish' => 32
34
+ 'Latvian' => 33
35
+ 'Estonian' => 34
36
+ 'Arabic' => 35
37
+ 'Frisian' => 36
38
+ 'Mangolian' => 37
39
+ 'Chinese_Taiwan' => 38
40
+ 'Indonesian' => 39
41
+ 'Maltese' => 40
42
+ 'Kabyle' => 41
43
+ 'Romanian' => 42
44
+ 'Chinese_China' => 43
45
+ 'German' => 44
46
  ================
47
  'starting_index' => 0