DylanonWic
commited on
Commit
•
e040936
1
Parent(s):
1428e4f
update model card README.md
Browse files
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 |
-
-
|
19 |
-
-
|
20 |
-
-
|
|
|
|
|
|
|
|
|
|
|
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.
|
40 |
-
- train_batch_size:
|
41 |
- eval_batch_size: 8
|
42 |
- seed: 42
|
43 |
- gradient_accumulation_steps: 3
|
44 |
-
- total_train_batch_size:
|
45 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
46 |
- lr_scheduler_type: linear
|
47 |
-
- lr_scheduler_warmup_steps:
|
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.
|
65 |
- Pytorch 1.13.1+cu116
|
66 |
-
- Datasets 2.
|
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
|