TheRamsay commited on
Commit
2531bd3
·
verified ·
1 Parent(s): 1b7ee5b

End of training

Browse files
Files changed (3) hide show
  1. README.md +86 -0
  2. generation_config.json +7 -0
  3. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,86 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ tags:
4
+ - generated_from_trainer
5
+ datasets:
6
+ - common_voice_17_0
7
+ metrics:
8
+ - wer
9
+ model-index:
10
+ - name: wav2vec2-gpt2-enc-dec
11
+ results:
12
+ - task:
13
+ name: Automatic Speech Recognition
14
+ type: automatic-speech-recognition
15
+ dataset:
16
+ name: common_voice_17_0
17
+ type: common_voice_17_0
18
+ config: cs
19
+ split: train+validation
20
+ args: cs
21
+ metrics:
22
+ - name: Wer
23
+ type: wer
24
+ value: 1.0114942528735633
25
+ ---
26
+
27
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
28
+ should probably proofread and complete it, then remove this comment. -->
29
+
30
+ # wav2vec2-gpt2-enc-dec
31
+
32
+ This model is a fine-tuned version of [](https://huggingface.co/) on the common_voice_17_0 dataset.
33
+ It achieves the following results on the evaluation set:
34
+ - Loss: 2.3875
35
+ - Wer: 1.0115
36
+
37
+ ## Model description
38
+
39
+ More information needed
40
+
41
+ ## Intended uses & limitations
42
+
43
+ More information needed
44
+
45
+ ## Training and evaluation data
46
+
47
+ More information needed
48
+
49
+ ## Training procedure
50
+
51
+ ### Training hyperparameters
52
+
53
+ The following hyperparameters were used during training:
54
+ - learning_rate: 0.0005
55
+ - train_batch_size: 16
56
+ - eval_batch_size: 8
57
+ - seed: 42
58
+ - gradient_accumulation_steps: 2
59
+ - total_train_batch_size: 32
60
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
61
+ - lr_scheduler_type: linear
62
+ - lr_scheduler_warmup_ratio: 0.08
63
+ - num_epochs: 20
64
+ - mixed_precision_training: Native AMP
65
+
66
+ ### Training results
67
+
68
+ | Training Loss | Epoch | Step | Validation Loss | Wer |
69
+ |:-------------:|:-------:|:-----:|:---------------:|:------:|
70
+ | 0.5699 | 2.1942 | 2000 | 0.5029 | 0.9499 |
71
+ | 0.4929 | 4.3884 | 4000 | 0.4351 | 0.9349 |
72
+ | 0.4568 | 6.5826 | 6000 | 0.3928 | 0.9080 |
73
+ | 0.4274 | 8.7767 | 8000 | 0.3524 | 0.8848 |
74
+ | 0.389 | 10.9709 | 10000 | 0.3127 | 0.8331 |
75
+ | 0.6396 | 13.1651 | 12000 | 0.5697 | 0.9324 |
76
+ | 2.5933 | 15.3593 | 14000 | 2.3935 | 1.0109 |
77
+ | 2.5858 | 17.5535 | 16000 | 2.3892 | 1.0120 |
78
+ | 2.5724 | 19.7477 | 18000 | 2.3875 | 1.0115 |
79
+
80
+
81
+ ### Framework versions
82
+
83
+ - Transformers 4.45.2
84
+ - Pytorch 2.4.1+cu121
85
+ - Datasets 3.0.1
86
+ - Tokenizers 0.20.0
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 50256,
4
+ "decoder_start_token_id": 50256,
5
+ "eos_token_id": 50256,
6
+ "transformers_version": "4.45.2"
7
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:69d20e69357d611dab3961e6c2409b9d728b35d62c61600d0ff5d3b499cc40ea
3
  size 988766816
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d60185d19f331518edb8857d17b658e8510bc3c7ff2e99bc00ccf00fe0a0bf0
3
  size 988766816