FYP19 commited on
Commit
6745564
1 Parent(s): c51b6d4

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -20
README.md CHANGED
@@ -14,10 +14,15 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [FYP19/t5-small-finetuned-wikisql](https://huggingface.co/FYP19/t5-small-finetuned-wikisql) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Loss: 0.0581
18
- - Rouge2 Precision: 0.6546
19
- - Rouge2 Recall: 0.4183
20
- - Rouge2 Fmeasure: 0.4794
 
 
 
 
 
21
 
22
  ## Model description
23
 
@@ -37,28 +42,16 @@ More information needed
37
 
38
  The following hyperparameters were used during training:
39
  - learning_rate: 5e-05
40
- - train_batch_size: 4
41
- - eval_batch_size: 4
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
- - lr_scheduler_warmup_steps: 500
46
- - num_epochs: 5
47
-
48
- ### Training results
49
-
50
- | Training Loss | Epoch | Step | Validation Loss | Rouge2 Precision | Rouge2 Recall | Rouge2 Fmeasure |
51
- |:-------------:|:-----:|:-----:|:---------------:|:----------------:|:-------------:|:---------------:|
52
- | 0.0446 | 1.0 | 5250 | 0.0595 | 0.5997 | 0.3824 | 0.4385 |
53
- | 0.0301 | 2.0 | 10500 | 0.0573 | 0.6439 | 0.4169 | 0.476 |
54
- | 0.0256 | 3.0 | 15750 | 0.0564 | 0.642 | 0.4093 | 0.4695 |
55
- | 0.0222 | 4.0 | 21000 | 0.0581 | 0.6487 | 0.4165 | 0.4769 |
56
- | 0.0218 | 5.0 | 26250 | 0.0581 | 0.6546 | 0.4183 | 0.4794 |
57
-
58
 
59
  ### Framework versions
60
 
61
  - Transformers 4.28.0
62
  - Pytorch 2.0.1+cu118
63
- - Datasets 2.12.0
64
  - Tokenizers 0.13.3
 
14
 
15
  This model is a fine-tuned version of [FYP19/t5-small-finetuned-wikisql](https://huggingface.co/FYP19/t5-small-finetuned-wikisql) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - eval_loss: 0.0595
18
+ - eval_rouge2_precision: 0.6531
19
+ - eval_rouge2_recall: 0.4156
20
+ - eval_rouge2_fmeasure: 0.4774
21
+ - eval_runtime: 13.7731
22
+ - eval_samples_per_second: 75.074
23
+ - eval_steps_per_second: 2.396
24
+ - epoch: 12.0
25
+ - step: 7884
26
 
27
  ## Model description
28
 
 
42
 
43
  The following hyperparameters were used during training:
44
  - learning_rate: 5e-05
45
+ - train_batch_size: 32
46
+ - eval_batch_size: 32
47
  - seed: 42
48
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
49
  - lr_scheduler_type: linear
50
+ - num_epochs: 20
 
 
 
 
 
 
 
 
 
 
 
 
51
 
52
  ### Framework versions
53
 
54
  - Transformers 4.28.0
55
  - Pytorch 2.0.1+cu118
56
+ - Datasets 2.14.4
57
  - Tokenizers 0.13.3