End of training

README.md

@@ -2,11 +2,26 @@
 library_name: transformers
 tags:
 - generated_from_trainer
+datasets:
+- common_voice_17_0
 metrics:
 - wer
 model-index:
 - name: wav2vec2-gpt2-enc-dec
-  results: []
+  results:
+  - task:
+      name: Automatic Speech Recognition
+      type: automatic-speech-recognition
+    dataset:
+      name: common_voice_17_0
+      type: common_voice_17_0
+      config: cs
+      split: train[:500]
+      args: cs
+    metrics:
+    - name: Wer
+      type: wer
+      value: 0.8489326765188834
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -14,10 +29,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # wav2vec2-gpt2-enc-dec
 
-This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
+This model is a fine-tuned version of [](https://huggingface.co/) on the common_voice_17_0 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 10.3257
-- Wer: 12.8392
+- Loss: 0.3276
+- Wer: 0.8489
 
 ## Model description
 
@@ -37,25 +52,36 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 1e-05
-- train_batch_size: 32
-- eval_batch_size: 32
+- train_batch_size: 16
+- eval_batch_size: 16
 - seed: 42
-- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- lr_scheduler_warmup_ratio: 0.3
-- num_epochs: 1
+- lr_scheduler_warmup_ratio: 0.08
+- num_epochs: 20
 - mixed_precision_training: Native AMP
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Wer     |
-|:-------------:|:-----:|:----:|:---------------:|:-------:|
-| 10.228        | 1.0   | 1    | 10.3257         | 12.8392 |
+| Training Loss | Epoch   | Step | Validation Loss | Wer    |
+|:-------------:|:-------:|:----:|:---------------:|:------:|
+| 1.9498        | 1.5625  | 50   | 0.6548          | 0.9324 |
+| 0.4531        | 3.125   | 100  | 0.3959          | 0.9020 |
+| 0.4087        | 4.6875  | 150  | 0.3735          | 0.8894 |
+| 0.3992        | 6.25    | 200  | 0.3572          | 0.8747 |
+| 0.3725        | 7.8125  | 250  | 0.3500          | 0.8763 |
+| 0.3635        | 9.375   | 300  | 0.3419          | 0.8626 |
+| 0.3647        | 10.9375 | 350  | 0.3381          | 0.8632 |
+| 0.36          | 12.5    | 400  | 0.3340          | 0.8566 |
+| 0.3588        | 14.0625 | 450  | 0.3316          | 0.8547 |
+| 0.362         | 15.625  | 500  | 0.3299          | 0.8547 |
+| 0.3613        | 17.1875 | 550  | 0.3280          | 0.8498 |
+| 0.3505        | 18.75   | 600  | 0.3276          | 0.8489 |
 
 
 ### Framework versions
 
-- Transformers 4.45.2
-- Pytorch 2.4.1+cu121
-- Datasets 3.0.1
-- Tokenizers 0.20.0
+- Transformers 4.48.3
+- Pytorch 2.6.0+cu124
+- Datasets 3.2.0
+- Tokenizers 0.21.0

generation_config.json

@@ -1,7 +1,6 @@
 {
   "_from_model_config": true,
-  "bos_token_id": 50256,
-  "decoder_start_token_id": 50256,
-  "eos_token_id": 50256,
-  "transformers_version": "4.45.2"
+  "bos_token_id": 0,
+  "eos_token_id": 0,
+  "transformers_version": "4.48.3"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:73efbe54ed0062fa1a41ad6743b63ee3995e5aa84a83dc01d5bab7c630cad1ba
+oid sha256:7119f49ce638161b35ac3584e689e52d18981c15d98abba04b69df2c2a9bd0ea
 size 550862672