update model card README.md

README.md

@@ -5,24 +5,9 @@ tags:
 - generated_from_trainer
 datasets:
 - PolyAI/minds14
-metrics:
-- wer
 model-index:
 - name: whisper-tiny-enUS
-  results:
-  - task:
-      name: Automatic Speech Recognition
-      type: automatic-speech-recognition
-    dataset:
-      name: PolyAI/minds14
-      type: PolyAI/minds14
-      config: en-US
-      split: train
-      args: en-US
-    metrics:
-    - name: Wer
-      type: wer
-      value: 7.431944109853047
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,9 +17,14 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on the PolyAI/minds14 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0000
-- Wer Ortho: 7.4972
-- Wer: 7.4319
+- eval_loss: 0.6151
+- eval_wer_ortho: 24.3412
+- eval_wer: 0.2421
+- eval_runtime: 9.0197
+- eval_samples_per_second: 12.417
+- eval_steps_per_second: 0.776
+- epoch: 35.71
+- step: 500
 
 ## Model description
 
@@ -53,32 +43,20 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 1e-05
+- learning_rate: 3e-05
 - train_batch_size: 16
 - eval_batch_size: 16
 - seed: 42
+- gradient_accumulation_steps: 2
+- total_train_batch_size: 32
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
-- lr_scheduler_type: constant_with_warmup
-- lr_scheduler_warmup_steps: 50
-- training_steps: 4000
-
-### Training results
-
-| Training Loss | Epoch  | Step | Validation Loss | Wer Ortho | Wer    |
-|:-------------:|:------:|:----:|:---------------:|:---------:|:------:|
-| 0.0017        | 14.29  | 500  | 0.0011          | 4.3200    | 4.2159 |
-| 0.0003        | 28.57  | 1000 | 0.0005          | 4.4204    | 4.3724 |
-| 0.001         | 42.86  | 1500 | 0.0003          | 4.1567    | 4.0954 |
-| 0.0001        | 57.14  | 2000 | 0.0001          | 4.3702    | 4.3483 |
-| 0.0001        | 71.43  | 2500 | 0.0001          | 7.1958    | 7.1429 |
-| 0.0           | 85.71  | 3000 | 0.0000          | 7.5097    | 7.4440 |
-| 0.0           | 100.0  | 3500 | 0.0000          | 7.5348    | 7.4681 |
-| 0.0           | 114.29 | 4000 | 0.0000          | 7.4972    | 7.4319 |
-
+- lr_scheduler_type: linear
+- lr_scheduler_warmup_steps: 100
+- training_steps: 5000
 
 ### Framework versions
 
-- Transformers 4.32.0.dev0
+- Transformers 4.31.0
 - Pytorch 2.0.1+cu117
 - Datasets 2.13.1
 - Tokenizers 0.13.3