Model save

README.md

@@ -2,8 +2,6 @@
 license: apache-2.0
 tags:
 - generated_from_trainer
-datasets:
-- wikisql
 model-index:
 - name: flan-t5-base-finetune
   results: []
@@ -14,12 +12,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 # flan-t5-base-finetune
 
-This model is a fine-tuned version of [google/flan-t5-base](https://huggingface.co/google/flan-t5-base) on the wikisql dataset.
+This model is a fine-tuned version of [htriedman/flan-t5-base-finetune](https://huggingface.co/htriedman/flan-t5-base-finetune) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0966
-- Rouge2 Precision: 0.8525
-- Rouge2 Recall: 0.5466
-- Rouge2 Fmeasure: 0.6549
+- Loss: 0.0175
+- Rouge2 Precision: 0.8828
+- Rouge2 Recall: 0.1209
+- Rouge2 Fmeasure: 0.2082
 
 ## Model description
 
@@ -39,8 +37,8 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
-- train_batch_size: 32
-- eval_batch_size: 128
+- train_batch_size: 4
+- eval_batch_size: 32
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
@@ -48,13 +46,13 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step  | Validation Loss | Rouge2 Precision | Rouge2 Recall | Rouge2 Fmeasure |
-|:-------------:|:-----:|:-----:|:---------------:|:----------------:|:-------------:|:---------------:|
-| 0.162         | 1.0   | 2025  | 0.1235          | 0.8357           | 0.5366        | 0.6425          |
-| 0.1277        | 2.0   | 4050  | 0.1082          | 0.8434           | 0.5413        | 0.6482          |
-| 0.1142        | 3.0   | 6075  | 0.1000          | 0.8491           | 0.5443        | 0.6522          |
-| 0.1063        | 4.0   | 8100  | 0.0978          | 0.8516           | 0.5464        | 0.6544          |
-| 0.1           | 5.0   | 10125 | 0.0966          | 0.8525           | 0.5466        | 0.6549          |
+| Training Loss | Epoch | Step   | Validation Loss | Rouge2 Precision | Rouge2 Recall | Rouge2 Fmeasure |
+|:-------------:|:-----:|:------:|:---------------:|:----------------:|:-------------:|:---------------:|
+| 0.0258        | 1.0   | 54375  | 0.0238          | 0.8678           | 0.119         | 0.2048          |
+| 0.0236        | 2.0   | 108750 | 0.0199          | 0.877            | 0.1203        | 0.2071          |
+| 0.0166        | 3.0   | 163125 | 0.0185          | 0.88             | 0.1206        | 0.2076          |
+| 0.0134        | 4.0   | 217500 | 0.0177          | 0.8824           | 0.1208        | 0.208           |
+| 0.0146        | 5.0   | 271875 | 0.0175          | 0.8828           | 0.1209        | 0.2082          |
 
 
 ### Framework versions

runs/May05_06-03-20_eec8524d6951/events.out.tfevents.1683266615.eec8524d6951.114662.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:856645bc6f861dfe69bb5f52976e181c478dee0d3233b2c9885e3bcd3627d2d7
-size 94138
+oid sha256:ad91d93099b43da193c3aee79dfe94fe22c9e0b97ccfa01163c588f940905c47
+size 94498