minhngca_lab1_finetuning_colab

README.md

@@ -22,7 +22,7 @@ model-index:
     metrics:
     - name: Bleu
       type: bleu
-      value: 5.442046712028784
+      value: 52.88398487672078
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +32,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [Helsinki-NLP/opus-mt-en-fr](https://huggingface.co/Helsinki-NLP/opus-mt-en-fr) on the kde4 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 4.7774
-- Bleu: 5.4420
+- Loss: 0.8556
+- Bleu: 52.8840
 
 ## Model description
 
@@ -58,7 +58,7 @@ 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: 1
+- num_epochs: 3
 - mixed_precision_training: Native AMP
 
 ### Training results

generation_config.json

@@ -11,5 +11,6 @@
   "max_length": 512,
   "num_beams": 4,
   "pad_token_id": 59513,
+  "renormalize_logits": true,
   "transformers_version": "4.35.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:25e67788978b30c6454ead22c87ef7ba67126fce7598d870eff85acbda002f85
+oid sha256:e8eb885ab5a1f76534225a57286ff338a881e95c7e16c3df8e4cb696a4803d5f
 size 298705768

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5251b3f0412dc8922164c41ce26bd594988b0f8b497a3f940ca29c6583e8508f
+oid sha256:ab3e87951d3eb698ee2344da97986c70a10acfdfda9a32cb77bf7895d17785de
 size 4792