Izen78 commited on
Commit
41a7776
1 Parent(s): 3d6e7c1

End of training

Browse files
Files changed (2) hide show
  1. README.md +7 -11
  2. generation_config.json +1 -1
README.md CHANGED
@@ -20,8 +20,8 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the custom dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.0000
24
- - Wer: 0.0
25
 
26
  ## Model description
27
 
@@ -47,23 +47,19 @@ The following hyperparameters were used during training:
47
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
  - lr_scheduler_type: linear
49
  - lr_scheduler_warmup_steps: 500
50
- - training_steps: 5000
51
  - mixed_precision_training: Native AMP
52
 
53
  ### Training results
54
 
55
- | Training Loss | Epoch | Step | Validation Loss | Wer |
56
- |:-------------:|:------:|:----:|:---------------:|:---:|
57
- | 0.0 | 1000.0 | 1000 | 0.0000 | 0.0 |
58
- | 0.0 | 2000.0 | 2000 | 0.0000 | 0.0 |
59
- | 0.0 | 3000.0 | 3000 | 0.0000 | 0.0 |
60
- | 0.0 | 4000.0 | 4000 | 0.0000 | 0.0 |
61
- | 0.0 | 5000.0 | 5000 | 0.0000 | 0.0 |
62
 
63
 
64
  ### Framework versions
65
 
66
- - Transformers 4.45.1
67
  - Pytorch 2.4.1+cu121
68
  - Datasets 3.0.1
69
  - Tokenizers 0.20.0
 
20
 
21
  This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the custom dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.9237
24
+ - Wer: 41.5842
25
 
26
  ## Model description
27
 
 
47
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
  - lr_scheduler_type: linear
49
  - lr_scheduler_warmup_steps: 500
50
+ - training_steps: 1000
51
  - mixed_precision_training: Native AMP
52
 
53
  ### Training results
54
 
55
+ | Training Loss | Epoch | Step | Validation Loss | Wer |
56
+ |:-------------:|:--------:|:----:|:---------------:|:-------:|
57
+ | 0.0 | 333.3333 | 1000 | 0.9237 | 41.5842 |
 
 
 
 
58
 
59
 
60
  ### Framework versions
61
 
62
+ - Transformers 4.45.2
63
  - Pytorch 2.4.1+cu121
64
  - Datasets 3.0.1
65
  - Tokenizers 0.20.0
generation_config.json CHANGED
@@ -250,5 +250,5 @@
250
  "transcribe": 50359,
251
  "translate": 50358
252
  },
253
- "transformers_version": "4.45.1"
254
  }
 
250
  "transcribe": 50359,
251
  "translate": 50358
252
  },
253
+ "transformers_version": "4.45.2"
254
  }