thuyentruong commited on
Commit
9658555
·
verified ·
1 Parent(s): a877f14

End of training

Browse files
Files changed (2) hide show
  1. README.md +29 -10
  2. model.safetensors +1 -1
README.md CHANGED
@@ -7,9 +7,24 @@ tags:
7
  - generated_from_trainer
8
  datasets:
9
  - PolyAI/minds14
 
 
10
  model-index:
11
  - name: Whisper tiny
12
- results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  ---
14
 
15
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -19,14 +34,9 @@ should probably proofread and complete it, then remove this comment. -->
19
 
20
  This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on the minds14 dataset.
21
  It achieves the following results on the evaluation set:
22
- - eval_loss: 0.5911
23
- - eval_wer_ortho: 0.3134
24
- - eval_wer: 0.3158
25
- - eval_runtime: 29.9977
26
- - eval_samples_per_second: 3.767
27
- - eval_steps_per_second: 0.5
28
- - epoch: 0.5
29
- - step: 28
30
 
31
  ## Model description
32
 
@@ -52,9 +62,18 @@ The following hyperparameters were used during training:
52
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
53
  - lr_scheduler_type: constant_with_warmup
54
  - lr_scheduler_warmup_steps: 8
55
- - training_steps: 225
56
  - mixed_precision_training: Native AMP
57
 
 
 
 
 
 
 
 
 
 
58
  ### Framework versions
59
 
60
  - Transformers 4.40.2
 
7
  - generated_from_trainer
8
  datasets:
9
  - PolyAI/minds14
10
+ metrics:
11
+ - wer
12
  model-index:
13
  - name: Whisper tiny
14
+ results:
15
+ - task:
16
+ name: Automatic Speech Recognition
17
+ type: automatic-speech-recognition
18
+ dataset:
19
+ name: minds14
20
+ type: PolyAI/minds14
21
+ config: en-US
22
+ split: train
23
+ args: en-US
24
+ metrics:
25
+ - name: Wer
26
+ type: wer
27
+ value: 0.3612750885478158
28
  ---
29
 
30
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
34
 
35
  This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on the minds14 dataset.
36
  It achieves the following results on the evaluation set:
37
+ - Loss: 0.6491
38
+ - Wer Ortho: 0.3572
39
+ - Wer: 0.3613
 
 
 
 
 
40
 
41
  ## Model description
42
 
 
62
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
63
  - lr_scheduler_type: constant_with_warmup
64
  - lr_scheduler_warmup_steps: 8
65
+ - training_steps: 45
66
  - mixed_precision_training: Native AMP
67
 
68
+ ### Training results
69
+
70
+ | Training Loss | Epoch | Step | Validation Loss | Wer Ortho | Wer |
71
+ |:-------------:|:------:|:----:|:---------------:|:---------:|:------:|
72
+ | 0.0569 | 0.2679 | 15 | 0.6113 | 0.3337 | 0.3294 |
73
+ | 0.0364 | 0.5357 | 30 | 0.6443 | 0.3603 | 0.3554 |
74
+ | 0.0916 | 0.8036 | 45 | 0.6491 | 0.3572 | 0.3613 |
75
+
76
+
77
  ### Framework versions
78
 
79
  - Transformers 4.40.2
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:56409b5b0400d2456dc41b706d2b96302ce0c3a9a899db6f0a5e57251aab72f3
3
  size 151061672
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:adbee272967c7d3ec1533f6cf37f5e6de7662b8114cdf7f9249530d8527fd7c8
3
  size 151061672