TeeA/T5-Text2SQL-Bilingual

README.md

@@ -2,6 +2,8 @@
 base_model: TeeA/T5-Text2SQL-Bilingual
 tags:
 - generated_from_trainer
+metrics:
+- rouge
 model-index:
 - name: Text2SQL-Bilingual
   results: []
@@ -14,16 +16,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [TeeA/T5-Text2SQL-Bilingual](https://huggingface.co/TeeA/T5-Text2SQL-Bilingual) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 0.6685
-- eval_rouge1: 0.9039
-- eval_rouge2: 0.8589
-- eval_rougeL: 0.9017
-- eval_rougeLsum: 0.9014
-- eval_runtime: 34.9375
-- eval_samples_per_second: 14.311
-- eval_steps_per_second: 3.578
-- epoch: 13.0
-- step: 57057
+- Loss: 0.6352
+- Rouge1: 0.8954
+- Rouge2: 0.8464
+- Rougel: 0.8923
+- Rougelsum: 0.8922
 
 ## Model description
 
@@ -48,9 +45,25 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 20
+- num_epochs: 10
 - mixed_precision_training: Native AMP
 
+### Training results
+
+| Training Loss | Epoch | Step  | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
+|:-------------:|:-----:|:-----:|:---------------:|:------:|:------:|:------:|:---------:|
+| 0.8096        | 1.0   | 4389  | 0.6355          | 0.8975 | 0.8473 | 0.8940 | 0.8941    |
+| 0.7862        | 2.0   | 8778  | 0.6368          | 0.8972 | 0.8489 | 0.8938 | 0.8939    |
+| 0.7791        | 3.0   | 13167 | 0.6368          | 0.8963 | 0.8469 | 0.8927 | 0.8925    |
+| 0.7792        | 4.0   | 17556 | 0.6369          | 0.8954 | 0.8464 | 0.8919 | 0.8917    |
+| 0.7859        | 5.0   | 21945 | 0.6356          | 0.8949 | 0.8448 | 0.8914 | 0.8912    |
+| 0.7812        | 6.0   | 26334 | 0.6354          | 0.8962 | 0.8468 | 0.8928 | 0.8928    |
+| 0.7813        | 7.0   | 30723 | 0.6359          | 0.8950 | 0.8451 | 0.8916 | 0.8913    |
+| 0.7695        | 8.0   | 35112 | 0.6356          | 0.8947 | 0.8458 | 0.8916 | 0.8915    |
+| 0.7842        | 9.0   | 39501 | 0.6350          | 0.8950 | 0.8463 | 0.8920 | 0.8918    |
+| 0.7724        | 10.0  | 43890 | 0.6352          | 0.8954 | 0.8464 | 0.8923 | 0.8922    |
+
+
 ### Framework versions
 
 - Transformers 4.38.2

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:30008e1dc800339fac11304740eee15d34f4bae9c81cd3ba5c0f7ae4f1cc7bac
+oid sha256:d2220baf5ef6450e7d69294bd31e5fa86ab8edb37805e8e1edac9b176a778ecd
 size 282739752

runs/Mar15_20-11-32_ab73af60f7b0/events.out.tfevents.1710533531.ab73af60f7b0.1425.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:00046ef1c80ec3bfc5c5650b2dcdf581fcd768a745d20e711cd13ab78025c7b9
+size 29737

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:67dd5c6963a6765d22c12a9ea2c8fc8accdbfa11b20206a8e113de20864861ce
+oid sha256:29849804be87ac161d60cd46e24b2767971f02a8db096ffbae1dc8c1010522b2
 size 5048