End of training

README.md

@@ -8,22 +8,22 @@ tags:
 metrics:
 - wer
 model-index:
-- name: Whisper base fine tuned full -  ashe194
+- name: Whisper base fine tuned full - ashe194
   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. -->
 
-[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/ashe194-700/whisper-base-fine-tuning/runs/c0hq4k9i)
-# Whisper base fine tuned full -  ashe194
+[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/ashe194-700/whisper-base-fine-tuning/runs/wjyvnxax)
+# Whisper base fine tuned full - ashe194
 
 This model is a fine-tuned version of [openai/whisper-base](https://huggingface.co/openai/whisper-base) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0036
-- Wer: 20.8151
-- Cer: 12.2694
-- Wer Ortho: 21.9602
+- Loss: 0.0034
+- Wer: 0.3422
+- Cer: 0.3439
+- Wer Ortho: 0.5124
 
 ## Model description
 
@@ -50,15 +50,16 @@ The following hyperparameters were used during training:
 - total_train_batch_size: 64
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 4
+- num_epochs: 3
 - mixed_precision_training: Native AMP
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Wer     | Cer     | Wer Ortho |
-|:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:---------:|
-| No log        | 1.0   | 67   | 0.0075          | 14.1037 | 8.5981  | 14.8657   |
-| No log        | 2.0   | 134  | 0.0036          | 20.8151 | 12.2694 | 21.9602   |
+| Training Loss | Epoch  | Step | Validation Loss | Wer    | Cer    | Wer Ortho |
+|:-------------:|:------:|:----:|:---------------:|:------:|:------:|:---------:|
+| No log        | 0.9935 | 76   | 0.0065          | 0.4277 | 0.2934 | 0.7385    |
+| No log        | 2.0    | 153  | 0.0035          | 0.3707 | 0.3717 | 0.5275    |
+| No log        | 2.9804 | 228  | 0.0034          | 0.3422 | 0.3439 | 0.5124    |
 
 
 ### Framework versions