update model card README.md

README.md

@@ -1,4 +1,10 @@
-
+---
+license: apache-2.0
+tags:
+- generated_from_trainer
+model-index:
+  name: bach-arb
+---
 
 <!-- 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. -->
@@ -7,8 +13,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [jonatasgrosman/wav2vec2-large-xlsr-53-german](https://huggingface.co/jonatasgrosman/wav2vec2-large-xlsr-53-german) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 8.2433
-- Wer: 1.2104
+- Loss: 2.9404
+- Wer: 0.6130
 
 ## Model description
 
@@ -36,29 +42,29 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 500
-- num_epochs: 320
+- num_epochs: 115
 - mixed_precision_training: Native AMP
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Wer    |
-|:-------------:|:-----:|:----:|:---------------:|:------:|
-| 6.8552        | 20.0  | 100  | 3.0998          | 1.0    |
-| 2.7154        | 40.0  | 200  | 3.1360          | 1.0029 |
-| 1.778         | 60.0  | 300  | 3.9075          | 1.0173 |
-| 0.8588        | 80.0  | 400  | 5.1158          | 1.1354 |
-| 0.4147        | 100.0 | 500  | 6.1905          | 1.0403 |
-| 0.21          | 120.0 | 600  | 6.7319          | 1.3026 |
-| 0.1141        | 140.0 | 700  | 6.8747          | 1.1556 |
-| 0.0713        | 160.0 | 800  | 7.4069          | 1.2911 |
-| 0.0544        | 180.0 | 900  | 7.2074          | 1.1729 |
-| 0.0427        | 200.0 | 1000 | 7.7323          | 1.2219 |
-| 0.0325        | 220.0 | 1100 | 8.0445          | 1.1354 |
-| 0.0257        | 240.0 | 1200 | 7.8804          | 1.1210 |
-| 0.0194        | 260.0 | 1300 | 8.1589          | 1.1470 |
-| 0.0164        | 280.0 | 1400 | 8.2274          | 1.1470 |
-| 0.0132        | 300.0 | 1500 | 8.2207          | 1.1527 |
-| 0.0124        | 320.0 | 1600 | 8.2433          | 1.2104 |
+| Training Loss | Epoch  | Step | Validation Loss | Wer    |
+|:-------------:|:------:|:----:|:---------------:|:------:|
+| 27.8653       | 7.14   | 100  | 3.1369          | 1.0    |
+| 2.5975        | 14.28  | 200  | 2.1223          | 0.9976 |
+| 1.2001        | 21.41  | 300  | 1.7455          | 0.8774 |
+| 0.5938        | 28.55  | 400  | 1.8534          | 0.7981 |
+| 0.4001        | 35.69  | 500  | 2.3318          | 0.7740 |
+| 0.2895        | 42.83  | 600  | 2.2214          | 0.7163 |
+| 0.1853        | 49.97  | 700  | 2.4841          | 0.7043 |
+| 0.1318        | 57.14  | 800  | 2.9749          | 0.7139 |
+| 0.1067        | 64.28  | 900  | 2.4759          | 0.7115 |
+| 0.0635        | 71.41  | 1000 | 2.6708          | 0.6635 |
+| 0.0515        | 78.55  | 1100 | 3.0593          | 0.6923 |
+| 0.0455        | 85.69  | 1200 | 2.9637          | 0.6587 |
+| 0.0329        | 92.83  | 1300 | 2.9837          | 0.6346 |
+| 0.0232        | 99.97  | 1400 | 2.9361          | 0.6178 |
+| 0.021         | 107.14 | 1500 | 2.9221          | 0.6010 |
+| 0.0193        | 114.28 | 1600 | 2.9404          | 0.6130 |
 
 
 ### Framework versions