Training complete

README.md

@@ -18,11 +18,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [Falconsai/text_summarization](https://huggingface.co/Falconsai/text_summarization) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.6288
-- Rouge1: 0.2455
-- Rouge2: 0.1187
-- Rougel: 0.203
-- Rougelsum: 0.2317
+- Loss: 1.6120
+- Rouge1: 0.2483
+- Rouge2: 0.1203
+- Rougel: 0.2055
+- Rougelsum: 0.2344
 
 ## Model description
 
@@ -42,8 +42,8 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5.6e-05
-- train_batch_size: 16
-- eval_batch_size: 16
+- train_batch_size: 8
+- eval_batch_size: 8
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
@@ -53,8 +53,8 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step  | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
 |:-------------:|:-----:|:-----:|:---------------:|:------:|:------:|:------:|:---------:|
-| 1.8226        | 1.0   | 17431 | 1.6400          | 0.2449 | 0.1178 | 0.2024 | 0.2311    |
-| 1.7775        | 2.0   | 34862 | 1.6288          | 0.2455 | 0.1187 | 0.203  | 0.2317    |
+| 1.8062        | 1.0   | 32691 | 1.6262          | 0.248  | 0.1198 | 0.2053 | 0.234     |
+| 1.7563        | 2.0   | 65382 | 1.6120          | 0.2483 | 0.1203 | 0.2055 | 0.2344    |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fc72b5ec1f37fa2aea8702e94ad359864c523e028fdd50e0beb68df0780b3e32
+oid sha256:bfc1b2a185e5574ed09d43a0b43fb3b1449d347adbb00205a39060708afa7fb6
 size 242041896