CYX2003 commited on
Commit
90e890a
1 Parent(s): 5b1e761

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +65 -65
README.md CHANGED
@@ -1,65 +1,65 @@
1
- ---
2
- license: apache-2.0
3
- base_model: openai/whisper-tiny.en
4
- tags:
5
- - generated_from_trainer
6
- metrics:
7
- - wer
8
- model-index:
9
- - name: whisper-tiny-en2
10
- results: []
11
- ---
12
-
13
- <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
- should probably proofread and complete it, then remove this comment. -->
15
-
16
- # whisper-tiny-en2
17
-
18
- This model is a fine-tuned version of [openai/whisper-tiny.en](https://huggingface.co/openai/whisper-tiny.en) on an unknown dataset.
19
- It achieves the following results on the evaluation set:
20
- - Loss: 0.9164
21
- - Wer Ortho: 33.1839
22
- - Wer: 33.7778
23
-
24
- ## Model description
25
-
26
- More information needed
27
-
28
- ## Intended uses & limitations
29
-
30
- More information needed
31
-
32
- ## Training and evaluation data
33
-
34
- More information needed
35
-
36
- ## Training procedure
37
-
38
- ### Training hyperparameters
39
-
40
- The following hyperparameters were used during training:
41
- - learning_rate: 1e-05
42
- - train_batch_size: 8
43
- - eval_batch_size: 16
44
- - seed: 42
45
- - gradient_accumulation_steps: 2
46
- - total_train_batch_size: 16
47
- - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
- - lr_scheduler_type: constant_with_warmup
49
- - lr_scheduler_warmup_steps: 50
50
- - training_steps: 600
51
- - mixed_precision_training: Native AMP
52
-
53
- ### Training results
54
-
55
- | Training Loss | Epoch | Step | Validation Loss | Wer Ortho | Wer |
56
- |:-------------:|:-----:|:----:|:---------------:|:---------:|:-------:|
57
- | 0.0008 | 125.0 | 500 | 0.9164 | 33.1839 | 33.7778 |
58
-
59
-
60
- ### Framework versions
61
-
62
- - Transformers 4.41.1
63
- - Pytorch 2.2.2+cu121
64
- - Datasets 2.19.1
65
- - Tokenizers 0.19.1
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: openai/whisper-tiny.en
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - wer
8
+ model-index:
9
+ - name: whisper-tiny-en2
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # whisper-tiny-en2
17
+
18
+ This model is a fine-tuned version of [openai/whisper-tiny.en](https://huggingface.co/openai/whisper-tiny.en) on [hf-internal-testing/librispeech_asr_dummy](https://huggingface.co/datasets/hf-internal-testing/librispeech_asr_dummy).
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.9164
21
+ - Wer Ortho: 33.1839
22
+ - Wer: 33.7778
23
+
24
+ ## Model description
25
+
26
+ It is fine-tuned version of whisper-tiny model for the audio/video transcription feature in [PersonAI](https://personaiweb.vercel.app)
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
39
+
40
+ The following hyperparameters were used during training:
41
+ - learning_rate: 1e-05
42
+ - train_batch_size: 8
43
+ - eval_batch_size: 16
44
+ - seed: 42
45
+ - gradient_accumulation_steps: 2
46
+ - total_train_batch_size: 16
47
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
+ - lr_scheduler_type: constant_with_warmup
49
+ - lr_scheduler_warmup_steps: 50
50
+ - training_steps: 600
51
+ - mixed_precision_training: Native AMP
52
+
53
+ ### Training results
54
+
55
+ | Training Loss | Epoch | Step | Validation Loss | Wer Ortho | Wer |
56
+ |:-------------:|:-----:|:----:|:---------------:|:---------:|:-------:|
57
+ | 0.0008 | 125.0 | 500 | 0.9164 | 33.1839 | 33.7778 |
58
+
59
+
60
+ ### Framework versions
61
+
62
+ - Transformers 4.41.1
63
+ - Pytorch 2.2.2+cu121
64
+ - Datasets 2.19.1
65
+ - Tokenizers 0.19.1