arroyadr commited on
Commit
37de673
1 Parent(s): 474ce6c

End of training

Browse files
Files changed (2) hide show
  1. README.md +10 -9
  2. pytorch_model.bin +1 -1
README.md CHANGED
@@ -22,7 +22,7 @@ model-index:
22
  metrics:
23
  - name: Wer
24
  type: wer
25
- value: 38.33004602235371
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,9 +32,9 @@ should probably proofread and complete it, then remove this comment. -->
32
 
33
  This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on the PolyAI/minds14 dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.8337
36
- - Wer Ortho: 38.7783
37
- - Wer: 38.3300
38
 
39
  ## Model description
40
 
@@ -53,20 +53,21 @@ More information needed
53
  ### Training hyperparameters
54
 
55
  The following hyperparameters were used during training:
56
- - learning_rate: 1e-05
57
  - train_batch_size: 16
58
  - eval_batch_size: 16
59
  - seed: 42
60
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
61
  - lr_scheduler_type: constant_with_warmup
62
  - lr_scheduler_warmup_steps: 50
63
- - training_steps: 500
64
 
65
  ### Training results
66
 
67
- | Training Loss | Epoch | Step | Validation Loss | Wer Ortho | Wer |
68
- |:-------------:|:-----:|:----:|:---------------:|:---------:|:-------:|
69
- | 0.0007 | 17.86 | 500 | 0.8337 | 38.7783 | 38.3300 |
 
70
 
71
 
72
  ### Framework versions
 
22
  metrics:
23
  - name: Wer
24
  type: wer
25
+ value: 0.44904667981591057
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
32
 
33
  This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on the PolyAI/minds14 dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 1.2178
36
+ - Wer Ortho: 0.4585
37
+ - Wer: 0.4490
38
 
39
  ## Model description
40
 
 
53
  ### Training hyperparameters
54
 
55
  The following hyperparameters were used during training:
56
+ - learning_rate: 0.0001
57
  - train_batch_size: 16
58
  - eval_batch_size: 16
59
  - seed: 42
60
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
61
  - lr_scheduler_type: constant_with_warmup
62
  - lr_scheduler_warmup_steps: 50
63
+ - training_steps: 800
64
 
65
  ### Training results
66
 
67
+ | Training Loss | Epoch | Step | Validation Loss | Wer Ortho | Wer |
68
+ |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|
69
+ | 0.024 | 14.29 | 400 | 1.0343 | 0.4255 | 0.4149 |
70
+ | 0.0215 | 28.57 | 800 | 1.2178 | 0.4585 | 0.4490 |
71
 
72
 
73
  ### Framework versions
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:98f8288d9eedb6beaeeed109b7b5a66a7d739810b8012222ff6ed76ca62eec92
3
  size 151099049
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:94b51817d65090142c1457e5a219f3e9c4a5b6e547486fd20b25d7190cf6d546
3
  size 151099049