parambharat commited on
Commit
3d926ba
·
1 Parent(s): 9c974c3

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -12
README.md CHANGED
@@ -14,13 +14,13 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [openai/whisper-base](https://huggingface.co/openai/whisper-base) on the None dataset.
16
  It achieves the following results on the evaluation set:
17
- - eval_loss: 0.2507
18
- - eval_wer: 24.5943
19
- - eval_runtime: 48.432
20
- - eval_samples_per_second: 2.065
21
- - eval_steps_per_second: 0.268
22
- - epoch: 0.2
23
- - step: 6000
24
 
25
  ## Model description
26
 
@@ -40,11 +40,9 @@ More information needed
40
 
41
  The following hyperparameters were used during training:
42
  - learning_rate: 1e-05
43
- - train_batch_size: 16
44
- - eval_batch_size: 8
45
  - seed: 42
46
- - gradient_accumulation_steps: 4
47
- - total_train_batch_size: 64
48
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
49
  - lr_scheduler_type: linear
50
  - lr_scheduler_warmup_steps: 1000
@@ -54,6 +52,6 @@ The following hyperparameters were used during training:
54
  ### Framework versions
55
 
56
  - Transformers 4.26.0.dev0
57
- - Pytorch 1.13.0
58
  - Datasets 2.7.1.dev0
59
  - Tokenizers 0.13.2
 
14
 
15
  This model is a fine-tuned version of [openai/whisper-base](https://huggingface.co/openai/whisper-base) on the None dataset.
16
  It achieves the following results on the evaluation set:
17
+ - eval_loss: 0.2411
18
+ - eval_wer: 23.0395
19
+ - eval_runtime: 38.027
20
+ - eval_samples_per_second: 2.63
21
+ - eval_steps_per_second: 0.105
22
+ - epoch: 0.05
23
+ - step: 6500
24
 
25
  ## Model description
26
 
 
40
 
41
  The following hyperparameters were used during training:
42
  - learning_rate: 1e-05
43
+ - train_batch_size: 64
44
+ - eval_batch_size: 32
45
  - seed: 42
 
 
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_steps: 1000
 
52
  ### Framework versions
53
 
54
  - Transformers 4.26.0.dev0
55
+ - Pytorch 1.13.0+cu117
56
  - Datasets 2.7.1.dev0
57
  - Tokenizers 0.13.2