End of training

README.md

@@ -17,10 +17,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [xlm-roberta-base](https://huggingface.co/xlm-roberta-base) on the wmt20_mlqe_task1 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4996
-- R Squared: 0.3350
-- Mae: 0.5152
-- Pearson R: 0.6754
+- Loss: 0.5476
+- R Squared: 0.2710
+- Mae: 0.5508
+- Pearson R: 0.6399
 
 ## Model description
 
@@ -45,15 +45,17 @@ 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: 3
+- num_epochs: 5
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | R Squared | Mae    | Pearson R |
 |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:---------:|
-| No log        | 1.0   | 438  | 0.4880          | 0.3504    | 0.5667 | 0.6411    |
-| 0.7379        | 2.0   | 876  | 0.4884          | 0.3498    | 0.5207 | 0.6772    |
-| 0.5367        | 3.0   | 1314 | 0.4996          | 0.3350    | 0.5152 | 0.6754    |
+| No log        | 1.0   | 438  | 0.4918          | 0.3453    | 0.5703 | 0.5987    |
+| 0.7759        | 2.0   | 876  | 0.4742          | 0.3687    | 0.5147 | 0.6788    |
+| 0.5858        | 3.0   | 1314 | 0.4901          | 0.3476    | 0.5239 | 0.6641    |
+| 0.4156        | 4.0   | 1752 | 0.4853          | 0.3539    | 0.5293 | 0.6553    |
+| 0.3151        | 5.0   | 2190 | 0.5476          | 0.2710    | 0.5508 | 0.6399    |
 
 
 ### Framework versions

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ffd47ddf0124d4c5ae60a06399bb66c0625ef38c30105060233273ed91afb8cd
+oid sha256:b61caa0a47694c84ece89070b757eb8c017422e66427ace4625ba20702c281ad
 size 1112246641