End of training

README.md

@@ -1,40 +1,24 @@
 ---
-language:
-- ro
 license: apache-2.0
 base_model: openai/whisper-medium
 tags:
-- hf-asr-leaderboard
 - generated_from_trainer
-datasets:
-- mozilla-foundation/common_voice_16_1
 metrics:
 - wer
 model-index:
-- name: Whisper Medium Ro - Sarbu Vlad
-  results:
-  - task:
-      name: Automatic Speech Recognition
-      type: automatic-speech-recognition
-    dataset:
-      name: Common Voice 16.1
-      type: mozilla-foundation/common_voice_16_1
-      args: 'config: ro, split: test'
-    metrics:
-    - name: Wer
-      type: wer
-      value: 12.451220618930938
+- name: Whisper_medium_ro_VladS_02_16_24_1500_steps_multi_gpu
+  results: []
 ---
 
 <!-- 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 Medium Ro - Sarbu Vlad
+# Whisper_medium_ro_VladS_02_16_24_1500_steps_multi_gpu
 
-This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on the Common Voice 16.1 dataset.
+This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1471
-- Wer: 12.4512
+- Loss: 0.1719
+- Wer: 12.1033
 
 ## Model description
 
@@ -63,20 +47,24 @@ The following hyperparameters were used during training:
 - total_eval_batch_size: 48
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- lr_scheduler_warmup_steps: 500
-- training_steps: 1500
+- lr_scheduler_warmup_steps: 250
+- training_steps: 2500
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Wer     |
 |:-------------:|:-----:|:----:|:---------------:|:-------:|
-| 0.3367        | 0.98  | 250  | 0.1873          | 16.5290 |
-| 0.1131        | 1.96  | 500  | 0.1383          | 13.7792 |
-| 0.0597        | 2.94  | 750  | 0.1306          | 12.3695 |
-| 0.0311        | 3.92  | 1000 | 0.1329          | 12.2758 |
-| 0.0168        | 4.9   | 1250 | 0.1410          | 12.5117 |
-| 0.0102        | 5.88  | 1500 | 0.1471          | 12.4512 |
+| 0.1563        | 0.98  | 250  | 0.1542          | 14.6716 |
+| 0.0933        | 1.96  | 500  | 0.1306          | 13.0714 |
+| 0.0428        | 2.94  | 750  | 0.1298          | 11.8886 |
+| 0.0243        | 3.92  | 1000 | 0.1353          | 12.0096 |
+| 0.0147        | 4.9   | 1250 | 0.1433          | 12.1064 |
+| 0.0083        | 5.88  | 1500 | 0.1572          | 12.2606 |
+| 0.0052        | 6.86  | 1750 | 0.1591          | 12.3090 |
+| 0.0037        | 7.84  | 2000 | 0.1665          | 12.0307 |
+| 0.0026        | 8.82  | 2250 | 0.1708          | 12.0549 |
+| 0.0021        | 9.8   | 2500 | 0.1719          | 12.1033 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3f18758c485947a7a93213995afbf3c7d4ea429df1ea82af3127967f56e5bcea
+oid sha256:0b8e3360d4bf15cfcdeed82edf8399fa8262b2aa056f3e295c48072f4c46c8d4
 size 1634047376

runs/Feb16_21-52-08_ubuntu-llama/events.out.tfevents.1708113139.ubuntu-llama.782673.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d37060564abde7dc6b3d5d0d9d262c6222a46c09da5cdd2170a40ac3288fb715
-size 24187
+oid sha256:d8e9f8d72f27bce41d7135f6c503e15fe3d68fc40b1cfa6275e908aec9dea521
+size 24541