Update README.md

README.md

@@ -9,14 +9,16 @@ base_model: google-t5/t5-base
 metrics:
 - exact_match
 model-index:
-  - name: juanfra218/text2sql
-    results:
-      - task:
-          type: text-to-sql
-        metrics:
-          - name: exact_match
-            type: exact_match
-            value: 0.4322
+- name: juanfra218/text2sql
+  results:
+  - task:
+      type: text-to-sql
+    metrics:
+    - name: exact_match
+      type: exact_match
+      value: 0.4322
+tags:
+- sql
 ---
 
 # Fine-Tuned Google T5 Model for Text to SQL Translation
@@ -35,9 +37,23 @@ A fine-tuned version of the Google T5 model, trained for the task of translating
 
 Currently working to implement PICARD (Parsing Incrementally for Constrained Auto-Regressive Decoding from Language Models) to improve the results of this model. More details can be found in the original [PICARD paper](https://arxiv.org/abs/2109.05093).
 
-## Results
+## Training Parameters
 
-Results are currently being evaluated and will be posted here soon.
+```
+training_args = Seq2SeqTrainingArguments(
+    output_dir="./results",
+    evaluation_strategy="epoch",
+    learning_rate=2e-5,
+    per_device_train_batch_size=8,
+    per_device_eval_batch_size=8,
+    weight_decay=0.01,
+    save_total_limit=3,
+    num_train_epochs=3,
+    predict_with_generate=True,
+    fp16=True,
+    push_to_hub=False,
+)
+```
 
 ## Usage