update model card README.md

README.md

@@ -1,41 +1,38 @@
 ---
-language:
-- eu
 license: apache-2.0
 tags:
-- whisper-event
 - generated_from_trainer
 datasets:
-- mozilla-foundation/common_voice_13_0
+- common_voice_13_0
 metrics:
 - wer
 model-index:
-- name: Whisper Small Basque
+- name: openai/whisper-medium
   results:
   - task:
       name: Automatic Speech Recognition
       type: automatic-speech-recognition
     dataset:
-      name: mozilla-foundation/common_voice_13_0 eu
-      type: mozilla-foundation/common_voice_13_0
+      name: common_voice_13_0
+      type: common_voice_13_0
       config: eu
       split: test
       args: eu
     metrics:
     - name: Wer
       type: wer
-      value: 18.775568066750374
+      value: 13.996111628660538
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# Whisper Small Basque
+# openai/whisper-medium
 
-This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the mozilla-foundation/common_voice_13_0 eu dataset.
+This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on the common_voice_13_0 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3812
-- Wer: 18.7756
+- Loss: 0.2256
+- Wer: 13.9961
 
 ## Model description
 
@@ -55,8 +52,8 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 1e-05
-- train_batch_size: 32
-- eval_batch_size: 16
+- train_batch_size: 4
+- eval_batch_size: 8
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
@@ -68,11 +65,11 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Wer     |
 |:-------------:|:-----:|:----:|:---------------:|:-------:|
-| 0.1413        | 2.04  | 1000 | 0.3178          | 22.0139 |
-| 0.0181        | 4.07  | 2000 | 0.3376          | 20.2864 |
-| 0.0044        | 7.02  | 3000 | 0.3603          | 18.8768 |
-| 0.0016        | 9.06  | 4000 | 0.3812          | 18.7756 |
-| 0.0012        | 12.01 | 5000 | 0.3914          | 18.8302 |
+| 0.4198        | 0.2   | 1000 | 0.4102          | 28.3487 |
+| 0.2547        | 0.4   | 2000 | 0.3142          | 21.6432 |
+| 0.2145        | 0.6   | 3000 | 0.2610          | 17.5159 |
+| 0.0828        | 1.14  | 4000 | 0.2388          | 15.3003 |
+| 0.0729        | 1.34  | 5000 | 0.2256          | 13.9961 |
 
 
 ### Framework versions

nohup.out

@@ -201080,3 +201080,12 @@ Training completed. Do not forget to share your model on huggingface.co/models =
 [INFO|configuration_utils.py:447] 2023-07-23 03:05:18,867 >> Configuration saved in ./config.json
 [INFO|modeling_utils.py:1693] 2023-07-23 03:05:23,030 >> Model weights saved in ./pytorch_model.bin
 [INFO|feature_extraction_utils.py:368] 2023-07-23 03:05:23,040 >> Feature extractor saved in ./preprocessor_config.json
+{'eval_loss': 0.2256375402212143, 'eval_wer': 13.996111628660538, 'eval_runtime': 1809.5302, 'eval_samples_per_second': 3.642, 'eval_steps_per_second': 0.455, 'epoch': 1.34}
+{'train_runtime': 15779.8251, 'train_samples_per_second': 1.267, 'train_steps_per_second': 0.317, 'train_loss': 0.2607154120922089, 'epoch': 1.34}
+
+   ca66a5d..029efe6  main -> main
+
+07/23/2023 03:05:43 - WARNING - huggingface_hub.repository - To https://huggingface.co/xezpeleta/whisper-medium-eu
+   ca66a5d..029efe6  main -> main
+
+