arroyadr commited on
Commit
72f2106
1 Parent(s): 5169ba3

End of training

Browse files
Files changed (2) hide show
  1. README.md +9 -9
  2. generation_config.json +1 -1
README.md CHANGED
@@ -22,7 +22,7 @@ model-index:
22
  metrics:
23
  - name: Wer
24
  type: wer
25
- value: 0.49506903353057197
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: 1.4859
36
- - Wer Ortho: 0.5045
37
- - Wer: 0.4951
38
 
39
  ## Model description
40
 
@@ -53,25 +53,25 @@ More information needed
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: 2000
64
 
65
  ### Training results
66
 
67
  | Training Loss | Epoch | Step | Validation Loss | Wer Ortho | Wer |
68
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|
69
- | 0.0084 | 71.43 | 2000 | 1.4859 | 0.5045 | 0.4951 |
70
 
71
 
72
  ### Framework versions
73
 
74
- - Transformers 4.32.0.dev0
75
  - Pytorch 2.0.1+cu118
76
  - Datasets 2.14.4
77
  - Tokenizers 0.13.3
 
22
  metrics:
23
  - name: Wer
24
  type: wer
25
+ value: 0.2868796736913664
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: 0.6398
36
+ - Wer Ortho: 0.2934
37
+ - Wer: 0.2869
38
 
39
  ## Model description
40
 
 
53
  ### Training hyperparameters
54
 
55
  The following hyperparameters were used during training:
56
+ - learning_rate: 1e-05
57
+ - train_batch_size: 32
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.0005 | 35.71 | 500 | 0.6398 | 0.2934 | 0.2869 |
70
 
71
 
72
  ### Framework versions
73
 
74
+ - Transformers 4.33.0.dev0
75
  - Pytorch 2.0.1+cu118
76
  - Datasets 2.14.4
77
  - Tokenizers 0.13.3
generation_config.json CHANGED
@@ -249,5 +249,5 @@
249
  "transcribe": 50359,
250
  "translate": 50358
251
  },
252
- "transformers_version": "4.32.0.dev0"
253
  }
 
249
  "transcribe": 50359,
250
  "translate": 50358
251
  },
252
+ "transformers_version": "4.33.0.dev0"
253
  }