ptotheun commited on
Commit
88cc451
1 Parent(s): 4ff56a7
README.md CHANGED
@@ -1,7 +1,7 @@
1
  ---
2
  library_name: transformers
3
  license: apache-2.0
4
- base_model: biodatlab/whisper-th-medium
5
  tags:
6
  - generated_from_trainer
7
  model-index:
@@ -14,10 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  # outs
16
 
17
- This model is a fine-tuned version of [biodatlab/whisper-th-medium](https://huggingface.co/biodatlab/whisper-th-medium) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.1143
20
- - Cer: 4.7054
21
 
22
  ## Model description
23
 
@@ -40,18 +40,27 @@ The following hyperparameters were used during training:
40
  - train_batch_size: 16
41
  - eval_batch_size: 16
42
  - seed: 42
43
- - gradient_accumulation_steps: 4
44
- - total_train_batch_size: 64
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
 
47
  - num_epochs: 10.0
48
- - mixed_precision_training: Native AMP
49
 
50
  ### Training results
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Cer |
53
  |:-------------:|:------:|:----:|:---------------:|:------:|
54
- | 0.0095 | 5.9259 | 200 | 0.1143 | 4.7054 |
 
 
 
 
 
 
 
 
 
55
 
56
 
57
  ### Framework versions
 
1
  ---
2
  library_name: transformers
3
  license: apache-2.0
4
+ base_model: biodatlab/whisper-th-medium-combined
5
  tags:
6
  - generated_from_trainer
7
  model-index:
 
14
 
15
  # outs
16
 
17
+ This model is a fine-tuned version of [biodatlab/whisper-th-medium-combined](https://huggingface.co/biodatlab/whisper-th-medium-combined) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.0874
20
+ - Cer: 5.5267
21
 
22
  ## Model description
23
 
 
40
  - train_batch_size: 16
41
  - eval_batch_size: 16
42
  - seed: 42
43
+ - gradient_accumulation_steps: 2.0
44
+ - total_train_batch_size: 32.0
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
+ - lr_scheduler_warmup_ratio: 0.1
48
  - num_epochs: 10.0
 
49
 
50
  ### Training results
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Cer |
53
  |:-------------:|:------:|:----:|:---------------:|:------:|
54
+ | No log | 0.9926 | 67 | 0.0899 | 4.9847 |
55
+ | 0.2506 | 2.0 | 135 | 0.0949 | 4.8051 |
56
+ | 0.085 | 2.9926 | 202 | 0.0822 | 4.3363 |
57
+ | 0.085 | 4.0 | 270 | 0.0841 | 5.2042 |
58
+ | 0.0456 | 4.9926 | 337 | 0.0896 | 9.6136 |
59
+ | 0.0392 | 6.0 | 405 | 0.0875 | 4.7087 |
60
+ | 0.0392 | 6.9926 | 472 | 0.0879 | 4.7320 |
61
+ | 0.0296 | 8.0 | 540 | 0.0874 | 5.3804 |
62
+ | 0.0276 | 8.9926 | 607 | 0.0860 | 5.2873 |
63
+ | 0.0276 | 9.9259 | 670 | 0.0874 | 5.5267 |
64
 
65
 
66
  ### Framework versions
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "biodatlab/whisper-th-medium",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "apply_spec_augment": true,
@@ -30,10 +30,10 @@
30
  "is_encoder_decoder": true,
31
  "mask_feature_length": 64,
32
  "mask_feature_min_masks": 0,
33
- "mask_feature_prob": 1e-05,
34
  "mask_time_length": 10,
35
  "mask_time_min_masks": 2,
36
- "mask_time_prob": 1e-05,
37
  "max_length": 448,
38
  "max_source_positions": 1500,
39
  "max_target_positions": 448,
 
1
  {
2
+ "_name_or_path": "biodatlab/whisper-th-medium-combined",
3
  "activation_dropout": 0.0,
4
  "activation_function": "gelu",
5
  "apply_spec_augment": true,
 
30
  "is_encoder_decoder": true,
31
  "mask_feature_length": 64,
32
  "mask_feature_min_masks": 0,
33
+ "mask_feature_prob": 0.1,
34
  "mask_time_length": 10,
35
  "mask_time_min_masks": 2,
36
+ "mask_time_prob": 0.1,
37
  "max_length": 448,
38
  "max_source_positions": 1500,
39
  "max_target_positions": 448,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b479c16a1bd5744691525dead15d42c9d7054c90ea2eac26808884d44634763f
3
  size 3055544304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3f217f32202aec544ca36e34d10bf1729140e94d4984c12151c203a65251216d
3
  size 3055544304
runs/Oct07_11-43-27_babb89e2e784/events.out.tfevents.1728301412.babb89e2e784.2316.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b1d2f89bc35e614fe06a484708037306fe7ccd9a1fff35d431d80ef394f74c0a
3
+ size 10485
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:558331cf01557fc487451746ec6d64cda2b2ddfe1bf821572d098225c7a4bbee
3
  size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc352b343ade27a40fed5ff8303139b0b86d7232406ab3b3c53ff2ccb38ea02d
3
  size 5304