devkyle commited on
Commit
1a62585
1 Parent(s): 5ceb31c

End of training

Browse files
Files changed (2) hide show
  1. README.md +6 -6
  2. generation_config.json +1 -1
README.md CHANGED
@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [openai/whisper-base](https://huggingface.co/openai/whisper-base) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.9605
22
- - Wer: 40.5810
23
 
24
  ## Model description
25
 
@@ -52,10 +52,10 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Wer |
54
  |:-------------:|:-----:|:----:|:---------------:|:-------:|
55
- | 0.151 | 10.0 | 500 | 0.7583 | 45.9840 |
56
- | 0.0226 | 20.0 | 1000 | 0.9046 | 43.0211 |
57
- | 0.0033 | 30.0 | 1500 | 0.9496 | 41.0893 |
58
- | 0.0006 | 40.0 | 2000 | 0.9605 | 40.5810 |
59
 
60
 
61
  ### Framework versions
 
18
 
19
  This model is a fine-tuned version of [openai/whisper-base](https://huggingface.co/openai/whisper-base) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.9478
22
+ - Wer: 40.9025
23
 
24
  ## Model description
25
 
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Wer |
54
  |:-------------:|:-----:|:----:|:---------------:|:-------:|
55
+ | 0.5893 | 10.0 | 500 | 0.7149 | 51.7322 |
56
+ | 0.0835 | 20.0 | 1000 | 0.8352 | 42.5328 |
57
+ | 0.0199 | 30.0 | 1500 | 0.9229 | 41.2373 |
58
+ | 0.0066 | 40.0 | 2000 | 0.9478 | 40.9025 |
59
 
60
 
61
  ### Framework versions
generation_config.json CHANGED
@@ -39,7 +39,7 @@
39
  ],
40
  "bos_token_id": 50257,
41
  "decoder_start_token_id": 50258,
42
- "dropout": 0.01,
43
  "eos_token_id": 50257,
44
  "is_multilingual": true,
45
  "lang_to_id": {
 
39
  ],
40
  "bos_token_id": 50257,
41
  "decoder_start_token_id": 50258,
42
+ "dropout": 0.1,
43
  "eos_token_id": 50257,
44
  "is_multilingual": true,
45
  "lang_to_id": {