Model save
Browse files- README.md +92 -0
- generation_config.json +116 -0
- 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:
|
3 |
size 151066444
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9657b5fab43f61ebe639922d36e66a66416ff171b2b9a9d1d8db83df5b5dbc4c
|
3 |
size 151066444
|