bigmorning commited on
Commit
559dd14
1 Parent(s): 506845a

Upload TFWhisperForConditionalGeneration

Browse files
Files changed (3) hide show
  1. README.md +11 -11
  2. config.json +14 -2
  3. tf_model.h5 +1 -1
README.md CHANGED
@@ -14,12 +14,12 @@ probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Train Loss: 1.2830
18
- - Train Accuracy: 0.0262
19
- - Train Wermet: 10.0196
20
- - Validation Loss: 1.1100
21
- - Validation Accuracy: 0.0279
22
- - Validation Wermet: 9.5683
23
  - Epoch: 4
24
 
25
  ## Model description
@@ -46,11 +46,11 @@ The following hyperparameters were used during training:
46
 
47
  | Train Loss | Train Accuracy | Train Wermet | Validation Loss | Validation Accuracy | Validation Wermet | Epoch |
48
  |:----------:|:--------------:|:------------:|:---------------:|:-------------------:|:-----------------:|:-----:|
49
- | 5.0901 | 0.0113 | 53.3790 | 4.4090 | 0.0122 | 42.3548 | 0 |
50
- | 4.3135 | 0.0127 | 42.3551 | 3.9430 | 0.0149 | 37.1045 | 1 |
51
- | 3.3458 | 0.0173 | 31.6069 | 2.3945 | 0.0222 | 25.5461 | 2 |
52
- | 1.9669 | 0.0232 | 13.7935 | 1.4966 | 0.0261 | 6.9562 | 3 |
53
- | 1.2830 | 0.0262 | 10.0196 | 1.1100 | 0.0279 | 9.5683 | 4 |
54
 
55
 
56
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Train Loss: 1.3529
18
+ - Train Accuracy: 0.0258
19
+ - Train Wermet: 17.4367
20
+ - Validation Loss: 1.1712
21
+ - Validation Accuracy: 0.0276
22
+ - Validation Wermet: 17.7695
23
  - Epoch: 4
24
 
25
  ## Model description
 
46
 
47
  | Train Loss | Train Accuracy | Train Wermet | Validation Loss | Validation Accuracy | Validation Wermet | Epoch |
48
  |:----------:|:--------------:|:------------:|:---------------:|:-------------------:|:-----------------:|:-----:|
49
+ | 5.1027 | 0.0113 | 52.5530 | 4.4267 | 0.0121 | 41.4796 | 0 |
50
+ | 4.3285 | 0.0126 | 38.6893 | 3.9835 | 0.0145 | 33.6050 | 1 |
51
+ | 3.4573 | 0.0168 | 30.7714 | 2.5568 | 0.0215 | 31.7559 | 2 |
52
+ | 2.0878 | 0.0226 | 20.5131 | 1.5738 | 0.0257 | 21.2159 | 3 |
53
+ | 1.3529 | 0.0258 | 17.4367 | 1.1712 | 0.0276 | 17.7695 | 4 |
54
 
55
 
56
  ### Framework versions
config.json CHANGED
@@ -8,7 +8,6 @@
8
  ],
9
  "attention_dropout": 0.0,
10
  "begin_suppress_tokens": [
11
- 220,
12
  50257
13
  ],
14
  "bos_token_id": 50257,
@@ -24,7 +23,20 @@
24
  "encoder_layerdrop": 0.0,
25
  "encoder_layers": 4,
26
  "eos_token_id": 50257,
27
- "forced_decoder_ids": null,
 
 
 
 
 
 
 
 
 
 
 
 
 
28
  "init_std": 0.02,
29
  "is_encoder_decoder": true,
30
  "mask_feature_length": 10,
 
8
  ],
9
  "attention_dropout": 0.0,
10
  "begin_suppress_tokens": [
 
11
  50257
12
  ],
13
  "bos_token_id": 50257,
 
23
  "encoder_layerdrop": 0.0,
24
  "encoder_layers": 4,
25
  "eos_token_id": 50257,
26
+ "forced_decoder_ids": [
27
+ [
28
+ 1,
29
+ 50289
30
+ ],
31
+ [
32
+ 2,
33
+ 50359
34
+ ],
35
+ [
36
+ 3,
37
+ 50363
38
+ ]
39
+ ],
40
  "init_std": 0.02,
41
  "is_encoder_decoder": true,
42
  "mask_feature_length": 10,
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b06045a8e59fcdeb830e56482067081e1d795db5c7f70af959ec40cc49e75508
3
  size 151253984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:260882734c193ba6fce88107d176985ba68f9b545afbe4ebea6f30eff03225cc
3
  size 151253984