anakib1 commited on
Commit
8b4da8e
·
verified ·
1 Parent(s): 8dd5483

Model save

Browse files
Files changed (3) hide show
  1. README.md +92 -0
  2. generation_config.json +116 -0
  3. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: openai/whisper-tiny
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - wer
8
+ model-index:
9
+ - name: whisper-multi-diar-wer
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # whisper-multi-diar-wer
17
+
18
+ This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on an unknown dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 10.0666
21
+ - Wer: 582.8864
22
+ - Cer: 168.6522
23
+ - Speech Scored: 693831
24
+ - Speech Miss: 52252
25
+ - Speech Falarm: 117030
26
+ - Speaker Miss: 52252
27
+ - Speaker Falarm: 117030
28
+ - Speaker Error: 187216
29
+ - Speaker Correct: 1437.5240
30
+ - Diarization Error: 356498
31
+ - Frames: 600
32
+ - Speaker Wide Frames: 746083
33
+ - Speech Scored Ratio: 1156.385
34
+ - Speech Miss Ratio: 87.0867
35
+ - Speech Falarm Ratio: 195.05
36
+ - Speaker Correct Ratio: 2.3959
37
+ - Speaker Miss Ratio: 0.0700
38
+ - Speaker Falarm Ratio: 0.1569
39
+ - Speaker Error Ratio: 0.2509
40
+ - Diarization Error Ratio: 0.4778
41
+
42
+ ## Model description
43
+
44
+ More information needed
45
+
46
+ ## Intended uses & limitations
47
+
48
+ More information needed
49
+
50
+ ## Training and evaluation data
51
+
52
+ More information needed
53
+
54
+ ## Training procedure
55
+
56
+ ### Training hyperparameters
57
+
58
+ The following hyperparameters were used during training:
59
+ - learning_rate: 1e-05
60
+ - train_batch_size: 24
61
+ - eval_batch_size: 32
62
+ - seed: 42
63
+ - gradient_accumulation_steps: 2
64
+ - total_train_batch_size: 48
65
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
66
+ - lr_scheduler_type: linear
67
+ - lr_scheduler_warmup_ratio: 0.05
68
+ - num_epochs: 10
69
+ - mixed_precision_training: Native AMP
70
+
71
+ ### Training results
72
+
73
+ | Training Loss | Epoch | Step | Validation Loss | Wer | Cer | Speech Scored | Speech Miss | Speech Falarm | Speaker Miss | Speaker Falarm | Speaker Error | Speaker Correct | Diarization Error | Frames | Speaker Wide Frames | Speech Scored Ratio | Speech Miss Ratio | Speech Falarm Ratio | Speaker Correct Ratio | Speaker Miss Ratio | Speaker Falarm Ratio | Speaker Error Ratio | Diarization Error Ratio |
74
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|:-------------:|:-----------:|:-------------:|:------------:|:--------------:|:-------------:|:---------------:|:-----------------:|:------:|:-------------------:|:-------------------:|:-----------------:|:-------------------:|:---------------------:|:------------------:|:--------------------:|:-------------------:|:-----------------------:|
75
+ | 11.3437 | 1.0 | 42 | 10.7905 | 574.9471 | 166.1650 | 743633 | 2450 | 150302 | 2450 | 150302 | 202641 | 1427.9773 | 355393 | 600 | 746083 | 1239.3883 | 4.0833 | 250.5033 | 2.3800 | 0.0033 | 0.2015 | 0.2716 | 0.4763 |
76
+ | 10.3627 | 2.0 | 84 | 10.4901 | 578.0875 | 167.1479 | 735121 | 10962 | 136397 | 10962 | 136397 | 201434 | 1433.1820 | 348793 | 600 | 746083 | 1225.2017 | 18.27 | 227.3283 | 2.3886 | 0.0147 | 0.1828 | 0.2700 | 0.4675 |
77
+ | 9.9444 | 3.0 | 126 | 10.3015 | 569.6851 | 166.4943 | 715188 | 30895 | 127221 | 30895 | 127221 | 194291 | 1435.5347 | 352407 | 600 | 746083 | 1191.98 | 51.4917 | 212.035 | 2.3926 | 0.0414 | 0.1705 | 0.2604 | 0.4723 |
78
+ | 9.7658 | 4.0 | 168 | 10.2071 | 572.0536 | 166.8688 | 706081 | 40002 | 122962 | 40002 | 122962 | 191357 | 1436.2147 | 354321 | 600 | 746083 | 1176.8017 | 66.67 | 204.9367 | 2.3937 | 0.0536 | 0.1648 | 0.2565 | 0.4749 |
79
+ | 9.5093 | 5.0 | 210 | 10.1640 | 572.3712 | 166.9189 | 703250 | 42833 | 121335 | 42833 | 121335 | 190255 | 1436.8813 | 354423 | 600 | 746083 | 1172.0833 | 71.3883 | 202.225 | 2.3948 | 0.0574 | 0.1626 | 0.2550 | 0.4750 |
80
+ | 9.3069 | 6.0 | 252 | 10.1287 | 573.2534 | 167.0644 | 700202 | 45881 | 119938 | 45881 | 119938 | 189349 | 1436.9886 | 355168 | 600 | 746083 | 1167.0033 | 76.4683 | 199.8967 | 2.3950 | 0.0615 | 0.1608 | 0.2538 | 0.4760 |
81
+ | 9.2209 | 7.0 | 294 | 10.1009 | 582.8864 | 168.6522 | 698009 | 48074 | 118866 | 48074 | 118866 | 188639 | 1437.1880 | 355579 | 600 | 746083 | 1163.3483 | 80.1233 | 198.11 | 2.3953 | 0.0644 | 0.1593 | 0.2528 | 0.4766 |
82
+ | 9.0761 | 8.0 | 336 | 10.0912 | 582.8864 | 168.6522 | 695719 | 50364 | 117834 | 50364 | 117834 | 187684 | 1437.6227 | 355882 | 600 | 746083 | 1159.5317 | 83.94 | 196.39 | 2.3960 | 0.0675 | 0.1579 | 0.2516 | 0.4770 |
83
+ | 8.9928 | 9.0 | 378 | 10.0654 | 582.8864 | 168.6522 | 694031 | 52052 | 117145 | 52052 | 117145 | 187295 | 1437.4753 | 356492 | 600 | 746083 | 1156.7183 | 86.7533 | 195.2417 | 2.3958 | 0.0698 | 0.1570 | 0.2510 | 0.4778 |
84
+ | 8.9674 | 10.0 | 420 | 10.0666 | 582.8864 | 168.6522 | 693831 | 52252 | 117030 | 52252 | 117030 | 187216 | 1437.5240 | 356498 | 600 | 746083 | 1156.385 | 87.0867 | 195.05 | 2.3959 | 0.0700 | 0.1569 | 0.2509 | 0.4778 |
85
+
86
+
87
+ ### Framework versions
88
+
89
+ - Transformers 4.36.2
90
+ - Pytorch 2.0.0
91
+ - Datasets 2.16.1
92
+ - Tokenizers 0.15.0
generation_config.json ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "begin_suppress_tokens": [
3
+ 220,
4
+ 50257
5
+ ],
6
+ "bos_token_id": 50257,
7
+ "decoder_start_token_id": 50258,
8
+ "eos_token_id": 50257,
9
+ "forced_decoder_ids": [
10
+ [
11
+ 1,
12
+ 50259
13
+ ],
14
+ [
15
+ 2,
16
+ 50359
17
+ ],
18
+ [
19
+ 3,
20
+ 50363
21
+ ]
22
+ ],
23
+ "max_length": 448,
24
+ "pad_token_id": 50257,
25
+ "suppress_tokens": [
26
+ 1,
27
+ 2,
28
+ 7,
29
+ 8,
30
+ 9,
31
+ 10,
32
+ 14,
33
+ 25,
34
+ 26,
35
+ 27,
36
+ 28,
37
+ 29,
38
+ 31,
39
+ 58,
40
+ 59,
41
+ 60,
42
+ 61,
43
+ 62,
44
+ 63,
45
+ 90,
46
+ 91,
47
+ 92,
48
+ 93,
49
+ 359,
50
+ 503,
51
+ 522,
52
+ 542,
53
+ 873,
54
+ 893,
55
+ 902,
56
+ 918,
57
+ 922,
58
+ 931,
59
+ 1350,
60
+ 1853,
61
+ 1982,
62
+ 2460,
63
+ 2627,
64
+ 3246,
65
+ 3253,
66
+ 3268,
67
+ 3536,
68
+ 3846,
69
+ 3961,
70
+ 4183,
71
+ 4667,
72
+ 6585,
73
+ 6647,
74
+ 7273,
75
+ 9061,
76
+ 9383,
77
+ 10428,
78
+ 10929,
79
+ 11938,
80
+ 12033,
81
+ 12331,
82
+ 12562,
83
+ 13793,
84
+ 14157,
85
+ 14635,
86
+ 15265,
87
+ 15618,
88
+ 16553,
89
+ 16604,
90
+ 18362,
91
+ 18956,
92
+ 20075,
93
+ 21675,
94
+ 22520,
95
+ 26130,
96
+ 26161,
97
+ 26435,
98
+ 28279,
99
+ 29464,
100
+ 31650,
101
+ 32302,
102
+ 32470,
103
+ 36865,
104
+ 42863,
105
+ 47425,
106
+ 49870,
107
+ 50254,
108
+ 50258,
109
+ 50358,
110
+ 50359,
111
+ 50360,
112
+ 50361,
113
+ 50362
114
+ ],
115
+ "transformers_version": "4.36.2"
116
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:061acfb7cc4577ab5921597397db8920e9904a860727d2a9fbbe7f88c787054e
3
  size 151066444
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9657b5fab43f61ebe639922d36e66a66416ff171b2b9a9d1d8db83df5b5dbc4c
3
  size 151066444