Update README.md

README.md

@@ -15,7 +15,7 @@ tags:
 
 This model is a fine-tuned version of VietAI/envit5-translation on the mt_eng_vietnamese dataset. It achieves the following results on the evaluation set:
 
-- Loss: 0.4565
+- Loss: 0.1988
   
 ## Model Details
 
@@ -27,9 +27,21 @@ This model is a fine-tuned version of VietAI/envit5-translation on the mt_eng_vi
 - **Model type:** T5
 - **Finetuned from model [optional]:** [VietAI/envit5-translation](https://huggingface.co/VietAI/envit5-translation)
 
-### Recommendations
-
-<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
-
-Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
-
+### Hyperparameters
+The following hyperparameters were used during training:
+- Repo: [simpleT5](https://github.com/Shivanandroy/simpleT5)
+- Dataset: [mt_eng_vietnamese](https://huggingface.co/datasets/mt_eng_vietnamese) + custom (~40%)
+```
+    model.train(
+        train_df=train_df,
+        eval_df=test_df,
+        source_max_token_len=128,
+        target_max_token_len=128,
+        batch_size=32,
+        max_epochs=15,
+        use_gpu=True,
+        save_only_last_epoch=True,
+        precision=16,
+        dataloader_num_workers=4
+    )
+```