DylanonWic commited on
Commit
d1faa3b
1 Parent(s): 9c9abd3

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -1
README.md CHANGED
@@ -1,6 +1,8 @@
1
  ---
2
  tags:
3
  - generated_from_trainer
 
 
4
  model-index:
5
  - name: wav2vec2-large-asr-th
6
  results: []
@@ -12,6 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
12
  # wav2vec2-large-asr-th
13
 
14
  This model was trained from scratch on the None dataset.
 
 
 
 
15
 
16
  ## Model description
17
 
@@ -38,10 +44,22 @@ The following hyperparameters were used during training:
38
  - total_train_batch_size: 48
39
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
40
  - lr_scheduler_type: linear
41
- - lr_scheduler_warmup_steps: 300
42
  - training_steps: 3000
43
  - mixed_precision_training: Native AMP
44
 
 
 
 
 
 
 
 
 
 
 
 
 
45
  ### Framework versions
46
 
47
  - Transformers 4.26.1
 
1
  ---
2
  tags:
3
  - generated_from_trainer
4
+ metrics:
5
+ - wer
6
  model-index:
7
  - name: wav2vec2-large-asr-th
8
  results: []
 
14
  # wav2vec2-large-asr-th
15
 
16
  This model was trained from scratch on the None dataset.
17
+ It achieves the following results on the evaluation set:
18
+ - Loss: 0.4214
19
+ - Wer: 0.3708
20
+ - Cer: 0.1236
21
 
22
  ## Model description
23
 
 
44
  - total_train_batch_size: 48
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
+ - lr_scheduler_warmup_steps: 100
48
  - training_steps: 3000
49
  - mixed_precision_training: Native AMP
50
 
51
+ ### Training results
52
+
53
+ | Training Loss | Epoch | Step | Validation Loss | Wer | Cer |
54
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|
55
+ | 0.6395 | 0.86 | 500 | 0.5238 | 0.4279 | 0.1463 |
56
+ | 0.4942 | 1.71 | 1000 | 0.5227 | 0.4188 | 0.1404 |
57
+ | 0.4195 | 2.57 | 1500 | 0.4984 | 0.4019 | 0.1344 |
58
+ | 0.514 | 3.42 | 2000 | 0.4713 | 0.3828 | 0.1305 |
59
+ | 0.4964 | 4.28 | 2500 | 0.4490 | 0.3780 | 0.1261 |
60
+ | 0.5175 | 5.14 | 3000 | 0.4214 | 0.3708 | 0.1236 |
61
+
62
+
63
  ### Framework versions
64
 
65
  - Transformers 4.26.1