DylanonWic commited on
Commit
e040936
1 Parent(s): 1428e4f

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -22
README.md CHANGED
@@ -1,8 +1,6 @@
1
  ---
2
  tags:
3
  - generated_from_trainer
4
- metrics:
5
- - wer
6
  model-index:
7
  - name: wav2vec2-large-asr-th-2
8
  results: []
@@ -15,9 +13,14 @@ should probably proofread and complete it, then remove this comment. -->
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.1961
19
- - Wer: 0.3488
20
- - Cer: 0.0967
 
 
 
 
 
21
 
22
  ## Model description
23
 
@@ -36,32 +39,21 @@ More information needed
36
  ### Training hyperparameters
37
 
38
  The following hyperparameters were used during training:
39
- - learning_rate: 0.0002
40
- - train_batch_size: 24
41
  - eval_batch_size: 8
42
  - seed: 42
43
  - gradient_accumulation_steps: 3
44
- - total_train_batch_size: 72
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
- - lr_scheduler_warmup_steps: 600
48
  - training_steps: 5000
49
  - mixed_precision_training: Native AMP
50
 
51
- ### Training results
52
-
53
- | Training Loss | Epoch | Step | Validation Loss | Wer | Cer |
54
- |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|
55
- | 0.5833 | 0.36 | 1000 | 0.2734 | 0.4104 | 0.1166 |
56
- | 0.5996 | 0.71 | 2000 | 0.2426 | 0.3957 | 0.1114 |
57
- | 0.4372 | 1.07 | 3000 | 0.2234 | 0.3716 | 0.1029 |
58
- | 0.4109 | 1.42 | 4000 | 0.2070 | 0.3570 | 0.0976 |
59
- | 0.3881 | 1.78 | 5000 | 0.1961 | 0.3488 | 0.0967 |
60
-
61
-
62
  ### Framework versions
63
 
64
- - Transformers 4.27.3
65
  - Pytorch 1.13.1+cu116
66
- - Datasets 2.10.1
67
  - Tokenizers 0.13.2
 
1
  ---
2
  tags:
3
  - generated_from_trainer
 
 
4
  model-index:
5
  - name: wav2vec2-large-asr-th-2
6
  results: []
 
13
 
14
  This model was trained from scratch on the None dataset.
15
  It achieves the following results on the evaluation set:
16
+ - eval_loss: 0.1819
17
+ - eval_wer: 0.3346
18
+ - eval_cer: 0.0910
19
+ - eval_runtime: 735.0436
20
+ - eval_samples_per_second: 13.822
21
+ - eval_steps_per_second: 1.728
22
+ - epoch: 0.95
23
+ - step: 4000
24
 
25
  ## Model description
26
 
 
39
  ### Training hyperparameters
40
 
41
  The following hyperparameters were used during training:
42
+ - learning_rate: 0.0001
43
+ - train_batch_size: 16
44
  - eval_batch_size: 8
45
  - seed: 42
46
  - gradient_accumulation_steps: 3
47
+ - total_train_batch_size: 48
48
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
49
  - lr_scheduler_type: linear
50
+ - lr_scheduler_warmup_steps: 500
51
  - training_steps: 5000
52
  - mixed_precision_training: Native AMP
53
 
 
 
 
 
 
 
 
 
 
 
 
54
  ### Framework versions
55
 
56
+ - Transformers 4.27.4
57
  - Pytorch 1.13.1+cu116
58
+ - Datasets 2.11.0
59
  - Tokenizers 0.13.2