DylanonWic commited on
Commit
f9d18d1
1 Parent(s): 7df3820

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +17 -10
README.md CHANGED
@@ -2,6 +2,8 @@
2
  license: apache-2.0
3
  tags:
4
  - generated_from_trainer
 
 
5
  model-index:
6
  - name: wav2vec2-large-asr-th-2
7
  results: []
@@ -14,14 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [facebook/wav2vec2-large-xlsr-53](https://huggingface.co/facebook/wav2vec2-large-xlsr-53) on the None dataset.
16
  It achieves the following results on the evaluation set:
17
- - eval_loss: 0.3239
18
- - eval_wer: 0.4763
19
- - eval_cer: 0.1331
20
- - eval_runtime: 749.146
21
- - eval_samples_per_second: 13.562
22
- - eval_steps_per_second: 1.695
23
- - epoch: 0.95
24
- - step: 4000
25
 
26
  ## Model description
27
 
@@ -48,10 +45,20 @@ The following hyperparameters were used during training:
48
  - total_train_batch_size: 48
49
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
50
  - lr_scheduler_type: linear
51
- - lr_scheduler_warmup_steps: 800
52
- - training_steps: 5000
53
  - mixed_precision_training: Native AMP
54
 
 
 
 
 
 
 
 
 
 
 
55
  ### Framework versions
56
 
57
  - Transformers 4.27.3
 
2
  license: apache-2.0
3
  tags:
4
  - generated_from_trainer
5
+ metrics:
6
+ - wer
7
  model-index:
8
  - name: wav2vec2-large-asr-th-2
9
  results: []
 
16
 
17
  This model is a fine-tuned version of [facebook/wav2vec2-large-xlsr-53](https://huggingface.co/facebook/wav2vec2-large-xlsr-53) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.3300
20
+ - Wer: 0.4924
21
+ - Cer: 0.1398
 
 
 
 
 
22
 
23
  ## Model description
24
 
 
45
  - total_train_batch_size: 48
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
+ - lr_scheduler_warmup_steps: 600
49
+ - training_steps: 4000
50
  - mixed_precision_training: Native AMP
51
 
52
+ ### Training results
53
+
54
+ | Training Loss | Epoch | Step | Validation Loss | Wer | Cer |
55
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|
56
+ | 1.2666 | 0.24 | 1000 | 0.7343 | 0.6934 | 0.2410 |
57
+ | 0.9305 | 0.47 | 2000 | 0.4504 | 0.6006 | 0.1764 |
58
+ | 0.8241 | 0.71 | 3000 | 0.3653 | 0.5278 | 0.1547 |
59
+ | 0.7788 | 0.95 | 4000 | 0.3300 | 0.4924 | 0.1398 |
60
+
61
+
62
  ### Framework versions
63
 
64
  - Transformers 4.27.3