jonatasgrosman commited on
Commit
3bb1bfa
1 Parent(s): b81e321

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +87 -0
README.md ADDED
@@ -0,0 +1,87 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_trainer
5
+ datasets:
6
+ - common_voice_11_0
7
+ metrics:
8
+ - wer
9
+ model-index:
10
+ - name: whisper-small-pt-cv11-v6
11
+ results:
12
+ - task:
13
+ name: Automatic Speech Recognition
14
+ type: automatic-speech-recognition
15
+ dataset:
16
+ name: common_voice_11_0
17
+ type: common_voice_11_0
18
+ config: pt
19
+ split: test
20
+ args: pt
21
+ metrics:
22
+ - name: Wer
23
+ type: wer
24
+ value: 12.080123266563945
25
+ ---
26
+
27
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
28
+ should probably proofread and complete it, then remove this comment. -->
29
+
30
+ # whisper-small-pt-cv11-v6
31
+
32
+ This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the common_voice_11_0 dataset.
33
+ It achieves the following results on the evaluation set:
34
+ - Loss: 0.2831
35
+ - Wer: 12.0801
36
+ - Cer: 4.8494
37
+
38
+ ## Model description
39
+
40
+ More information needed
41
+
42
+ ## Intended uses & limitations
43
+
44
+ More information needed
45
+
46
+ ## Training and evaluation data
47
+
48
+ More information needed
49
+
50
+ ## Training procedure
51
+
52
+ ### Training hyperparameters
53
+
54
+ The following hyperparameters were used during training:
55
+ - learning_rate: 5e-06
56
+ - train_batch_size: 32
57
+ - eval_batch_size: 16
58
+ - seed: 42
59
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
60
+ - lr_scheduler_type: linear
61
+ - lr_scheduler_warmup_steps: 1000
62
+ - training_steps: 10000
63
+ - mixed_precision_training: Native AMP
64
+
65
+ ### Training results
66
+
67
+ | Training Loss | Epoch | Step | Validation Loss | Wer | Cer |
68
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:------:|
69
+ | 0.5235 | 0.92 | 500 | 0.3605 | 15.4083 | 6.1205 |
70
+ | 0.3839 | 1.84 | 1000 | 0.3034 | 14.2835 | 5.6010 |
71
+ | 0.2828 | 2.76 | 1500 | 0.2852 | 13.4977 | 5.1727 |
72
+ | 0.2367 | 3.68 | 2000 | 0.2768 | 12.9122 | 5.2280 |
73
+ | 0.1832 | 4.6 | 2500 | 0.2728 | 12.2496 | 4.9157 |
74
+ | 0.1549 | 5.52 | 3000 | 0.2730 | 12.0647 | 4.8384 |
75
+ | 0.1318 | 6.45 | 3500 | 0.2757 | 12.0955 | 4.8135 |
76
+ | 0.1077 | 7.37 | 4000 | 0.2738 | 11.9723 | 4.8273 |
77
+ | 0.0969 | 8.29 | 4500 | 0.2784 | 12.1572 | 4.9212 |
78
+ | 0.0813 | 9.21 | 5000 | 0.2805 | 12.3112 | 5.0207 |
79
+ | 0.0751 | 10.13 | 5500 | 0.2831 | 12.0801 | 4.8494 |
80
+
81
+
82
+ ### Framework versions
83
+
84
+ - Transformers 4.26.0.dev0
85
+ - Pytorch 1.12.1+cu116
86
+ - Datasets 2.7.1.dev0
87
+ - Tokenizers 0.13.2