mitro99 commited on
Commit
6c49cb7
1 Parent(s): 585e81e

End of training

Browse files
Files changed (3) hide show
  1. README.md +11 -2
  2. generation_config.json +5 -1
  3. model.safetensors +1 -1
README.md CHANGED
@@ -16,6 +16,15 @@ should probably proofread and complete it, then remove this comment. -->
16
  # whisper-tiny-polyai-enUS
17
 
18
  This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on the PolyAI/minds14 dataset.
 
 
 
 
 
 
 
 
 
19
 
20
  ## Model description
21
 
@@ -35,8 +44,8 @@ More information needed
35
 
36
  The following hyperparameters were used during training:
37
  - learning_rate: 1e-05
38
- - train_batch_size: 16
39
- - eval_batch_size: 16
40
  - seed: 42
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: constant_with_warmup
 
16
  # whisper-tiny-polyai-enUS
17
 
18
  This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on the PolyAI/minds14 dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - eval_loss: 0.7582
21
+ - eval_wer_ortho: 0.3510
22
+ - eval_wer: 0.3418
23
+ - eval_runtime: 34.041
24
+ - eval_samples_per_second: 3.32
25
+ - eval_steps_per_second: 0.118
26
+ - epoch: 26.67
27
+ - step: 400
28
 
29
  ## Model description
30
 
 
44
 
45
  The following hyperparameters were used during training:
46
  - learning_rate: 1e-05
47
+ - train_batch_size: 32
48
+ - eval_batch_size: 32
49
  - seed: 42
50
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
  - lr_scheduler_type: constant_with_warmup
generation_config.json CHANGED
@@ -35,11 +35,15 @@
35
  "forced_decoder_ids": [
36
  [
37
  1,
38
- null
39
  ],
40
  [
41
  2,
42
  50359
 
 
 
 
43
  ]
44
  ],
45
  "is_multilingual": true,
 
35
  "forced_decoder_ids": [
36
  [
37
  1,
38
+ 50259
39
  ],
40
  [
41
  2,
42
  50359
43
+ ],
44
+ [
45
+ 3,
46
+ 50363
47
  ]
48
  ],
49
  "is_multilingual": true,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6789410cfcf94bb4a5caf8047dfce105e1bd46f53d86a020f2c508573beb6950
3
  size 151061672
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fe990aa13e97ce1da240b938fec3da448b20c1509b4ef3729a795906104026d3
3
  size 151061672