update model card README.md

README.md

@@ -1,6 +1,8 @@
 ---
 tags:
 - generated_from_trainer
+metrics:
+- wer
 model-index:
 - name: wav2vec2-large-asr-th-2
   results: []
@@ -13,14 +15,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model was trained from scratch on the None dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 0.1819
-- eval_wer: 0.3346
-- eval_cer: 0.0910
-- eval_runtime: 735.0436
-- eval_samples_per_second: 13.822
-- eval_steps_per_second: 1.728
-- epoch: 0.95
-- step: 4000
+- Loss: 0.2501
+- Wer: 0.3205
+- Cer: 0.0879
 
 ## Model description
 
@@ -39,21 +36,31 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.0001
-- train_batch_size: 16
+- learning_rate: 2e-05
+- train_batch_size: 12
 - eval_batch_size: 8
 - seed: 42
 - gradient_accumulation_steps: 3
-- total_train_batch_size: 48
+- total_train_batch_size: 36
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 500
-- training_steps: 5000
+- training_steps: 4000
 - mixed_precision_training: Native AMP
 
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Wer    | Cer    |
+|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|
+| 0.1461        | 0.18  | 1000 | 0.4394          | 0.3238 | 0.0888 |
+| 0.1817        | 0.36  | 2000 | 0.4112          | 0.3261 | 0.0895 |
+| 0.2881        | 0.53  | 3000 | 0.3429          | 0.3241 | 0.0890 |
+| 0.7837        | 0.71  | 4000 | 0.2501          | 0.3205 | 0.0879 |
+
+
 ### Framework versions
 
-- Transformers 4.27.4
-- Pytorch 1.13.1+cu116
-- Datasets 2.11.0
-- Tokenizers 0.13.2
+- Transformers 4.28.1
+- Pytorch 2.0.0+cu118
+- Datasets 2.12.0
+- Tokenizers 0.13.3