geninhu commited on
Commit
d2f4d5b
1 Parent(s): ae984db

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -10
README.md CHANGED
@@ -16,9 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.6150
20
- - Wer: 20.4979
21
- - Cer: 8.9847
22
 
23
  ## Model description
24
 
@@ -37,10 +37,12 @@ More information needed
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
- - learning_rate: 1e-06
41
  - train_batch_size: 32
42
- - eval_batch_size: 16
43
  - seed: 42
 
 
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
  - lr_scheduler_warmup_steps: 500
@@ -51,11 +53,11 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Wer | Cer |
53
  |:-------------:|:-----:|:----:|:---------------:|:-------:|:------:|
54
- | 0.1041 | 0.2 | 1000 | 0.5133 | 22.6146 | 9.7131 |
55
- | 0.074 | 1.11 | 2000 | 0.5532 | 21.6166 | 9.4196 |
56
- | 0.0796 | 2.02 | 3000 | 0.6025 | 21.3314 | 9.2435 |
57
- | 0.0422 | 2.22 | 4000 | 0.6029 | 20.7392 | 9.0274 |
58
- | 0.0517 | 3.13 | 5000 | 0.6150 | 20.4979 | 8.9847 |
59
 
60
 
61
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.7427
20
+ - Wer: 19.3902
21
+ - Cer: 8.7285
22
 
23
  ## Model description
24
 
 
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
+ - learning_rate: 1e-05
41
  - train_batch_size: 32
42
+ - eval_batch_size: 32
43
  - seed: 42
44
+ - gradient_accumulation_steps: 2
45
+ - total_train_batch_size: 64
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_steps: 500
 
53
 
54
  | Training Loss | Epoch | Step | Validation Loss | Wer | Cer |
55
  |:-------------:|:-----:|:----:|:---------------:|:-------:|:------:|
56
+ | 0.0471 | 1.05 | 1000 | 0.5961 | 20.2895 | 8.9820 |
57
+ | 0.0194 | 2.11 | 2000 | 1.0999 | 22.6146 | 9.7105 |
58
+ | 0.002 | 4.02 | 3000 | 0.7289 | 20.2018 | 8.8379 |
59
+ | 0.0006 | 5.07 | 4000 | 0.7791 | 19.3683 | 8.4376 |
60
+ | 0.0003 | 6.13 | 5000 | 0.7427 | 19.3902 | 8.7285 |
61
 
62
 
63
  ### Framework versions