End of training

README.md

@@ -1,18 +1,18 @@
 ---
+license: apache-2.0
 base_model: facebook/wav2vec2-base
+tags:
+- generated_from_trainer
 datasets:
 - vivos
-license: apache-2.0
 metrics:
 - wer
-tags:
-- generated_from_trainer
 model-index:
 - name: wav2vec2-vivos
   results:
   - task:
-      type: automatic-speech-recognition
       name: Automatic Speech Recognition
+      type: automatic-speech-recognition
     dataset:
       name: vivos
       type: vivos
@@ -20,9 +20,9 @@ model-index:
       split: None
       args: default
     metrics:
-    - type: wer
-      value: 0.23636599442318915
-      name: Wer
+    - name: Wer
+      type: wer
+      value: 0.2342930262316059
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +32,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the vivos dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4755
-- Wer: 0.2364
+- Loss: 0.4598
+- Wer: 0.2343
 
 ## Model description
 
@@ -60,7 +60,7 @@ The following hyperparameters were used during training:
 - total_train_batch_size: 128
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- lr_scheduler_warmup_ratio: 0.2
+- lr_scheduler_warmup_ratio: 0.25
 - num_epochs: 20
 - mixed_precision_training: Native AMP
 
@@ -68,16 +68,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Wer    |
 |:-------------:|:-----:|:----:|:---------------:|:------:|
-| 1.349         | 2.0   | 146  | 0.9904          | 0.6088 |
-| 0.718         | 4.0   | 292  | 0.6959          | 0.4630 |
-| 0.4692        | 6.0   | 438  | 0.5304          | 0.3414 |
-| 0.3385        | 8.0   | 584  | 0.5078          | 0.3216 |
-| 0.2627        | 10.0  | 730  | 0.4659          | 0.2788 |
-| 0.2033        | 12.0  | 876  | 0.4751          | 0.2656 |
-| 0.1699        | 14.0  | 1022 | 0.4659          | 0.2519 |
-| 0.1688        | 16.0  | 1168 | 0.4662          | 0.2394 |
-| 0.1269        | 18.0  | 1314 | 0.4707          | 0.2375 |
-| 0.1162        | 20.0  | 1460 | 0.4755          | 0.2364 |
+| 6.8271        | 2.0   | 146  | 3.8747          | 1.0    |
+| 3.4616        | 4.0   | 292  | 3.5849          | 1.0    |
+| 3.35          | 6.0   | 438  | 2.6294          | 0.9997 |
+| 1.1993        | 8.0   | 584  | 0.6472          | 0.4255 |
+| 0.4734        | 10.0  | 730  | 0.5342          | 0.3258 |
+| 0.3156        | 12.0  | 876  | 0.4651          | 0.2758 |
+| 0.2392        | 14.0  | 1022 | 0.4690          | 0.2573 |
+| 0.2183        | 16.0  | 1168 | 0.4601          | 0.2434 |
+| 0.164         | 18.0  | 1314 | 0.4619          | 0.2379 |
+| 0.1452        | 20.0  | 1460 | 0.4598          | 0.2343 |
 
 
 ### Framework versions