velocity-ai commited on
Commit
3c2ef63
·
verified ·
1 Parent(s): ab2f10d

End of training

Browse files
README.md CHANGED
@@ -7,22 +7,9 @@ tags:
7
  - generated_from_trainer
8
  datasets:
9
  - velocity-whisper
10
- metrics:
11
- - wer
12
  model-index:
13
  - name: whisper-small-finetuned-hinglish
14
- results:
15
- - task:
16
- name: Automatic Speech Recognition
17
- type: automatic-speech-recognition
18
- dataset:
19
- name: exclamation
20
- type: velocity-whisper
21
- args: 'config: hi, split: test'
22
- metrics:
23
- - name: Wer
24
- type: wer
25
- value: 15.600350569675722
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,9 +18,6 @@ should probably proofread and complete it, then remove this comment. -->
31
  # whisper-small-finetuned-hinglish
32
 
33
  This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on the exclamation dataset.
34
- It achieves the following results on the evaluation set:
35
- - Loss: 0.1456
36
- - Wer: 15.6004
37
 
38
  ## Model description
39
 
@@ -59,19 +43,16 @@ The following hyperparameters were used during training:
59
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
60
  - lr_scheduler_type: linear
61
  - lr_scheduler_warmup_steps: 500
62
- - num_epochs: 20
63
  - mixed_precision_training: Native AMP
64
 
65
  ### Training results
66
 
67
- | Training Loss | Epoch | Step | Validation Loss | Wer |
68
- |:-------------:|:-------:|:----:|:---------------:|:-------:|
69
- | 0.0001 | 17.8571 | 1000 | 0.1456 | 15.6004 |
70
 
71
 
72
  ### Framework versions
73
 
74
- - Transformers 4.41.1
75
  - Pytorch 2.3.0+cu121
76
  - Datasets 2.19.1
77
  - Tokenizers 0.19.1
 
7
  - generated_from_trainer
8
  datasets:
9
  - velocity-whisper
 
 
10
  model-index:
11
  - name: whisper-small-finetuned-hinglish
12
+ results: []
 
 
 
 
 
 
 
 
 
 
 
13
  ---
14
 
15
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
18
  # whisper-small-finetuned-hinglish
19
 
20
  This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on the exclamation dataset.
 
 
 
21
 
22
  ## Model description
23
 
 
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
  - lr_scheduler_warmup_steps: 500
46
+ - num_epochs: 2
47
  - mixed_precision_training: Native AMP
48
 
49
  ### Training results
50
 
 
 
 
51
 
52
 
53
  ### Framework versions
54
 
55
+ - Transformers 4.41.2
56
  - Pytorch 2.3.0+cu121
57
  - Datasets 2.19.1
58
  - Tokenizers 0.19.1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "openai/whisper-medium",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "apply_spec_augment": false,
@@ -147,7 +147,7 @@
147
  50362
148
  ],
149
  "torch_dtype": "float32",
150
- "transformers_version": "4.41.1",
151
  "use_cache": true,
152
  "use_weighted_layer_sum": false,
153
  "vocab_size": 51865
 
1
  {
2
+ "_name_or_path": "velocity-ai/velocity-whisper-hi",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "apply_spec_augment": false,
 
147
  50362
148
  ],
149
  "torch_dtype": "float32",
150
+ "transformers_version": "4.41.2",
151
  "use_cache": true,
152
  "use_weighted_layer_sum": false,
153
  "vocab_size": 51865
generation_config.json CHANGED
@@ -236,5 +236,5 @@
236
  "transcribe": 50359,
237
  "translate": 50358
238
  },
239
- "transformers_version": "4.41.1"
240
  }
 
236
  "transcribe": 50359,
237
  "translate": 50358
238
  },
239
+ "transformers_version": "4.41.2"
240
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:61e6bda9514e492769a3f230d09105604064d2a6352ec60a3dca73ab32dba228
3
  size 3055544304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a92215cccedc7823ab5a7cb1bf75100be9cd826d32df33f52be459bd52e5205
3
  size 3055544304
runs/May31_13-47-08_e4d7e8146ce9/events.out.tfevents.1717163267.e4d7e8146ce9.992.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:682b337c282cd97d42426c2fa1ac758744fd7a98573b79a08f878fb9c37e87e8
3
+ size 12442
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:07d931df9884dc72d8e1e1d1a0975df5cae874973b5d7a52b4d5b5a1a9984e65
3
  size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:20fc80feac085c40fb2c767f695870e61acf0cb3321a7d589820e8b7e4ebdc45
3
  size 5240