Training complete

README.md

@@ -1,5 +1,5 @@
 ---
-base_model: october-sd/pegasus-xsum-finetuned-en-sum-2
+base_model: october-sd/pegasus-xsum-finetuned-en-sum
 tags:
 - summarization
 - generated_from_trainer
@@ -13,7 +13,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # pegasus-xsum-finetuned-en-sum
 
-This model is a fine-tuned version of [october-sd/pegasus-xsum-finetuned-en-sum-2](https://huggingface.co/october-sd/pegasus-xsum-finetuned-en-sum-2) on an unknown dataset.
+This model is a fine-tuned version of [october-sd/pegasus-xsum-finetuned-en-sum](https://huggingface.co/october-sd/pegasus-xsum-finetuned-en-sum) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 1.5095
 
 ## Model description
 
@@ -40,13 +42,17 @@ The following hyperparameters were used during training:
 - total_train_batch_size: 128
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 1
+- lr_scheduler_warmup_steps: 500
+- num_epochs: 4
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
 | No log        | 1.0   | 507  | 1.4958          |
+| No log        | 2.0   | 1014 | 1.5115          |
+| No log        | 3.0   | 1522 | 1.5094          |
+| No log        | 4.0   | 2028 | 1.5095          |
 
 
 ### Framework versions

runs/Apr05_18-30-01_863dff04394a/events.out.tfevents.1712341822.863dff04394a.34.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:72af275fc5c4a99475f78c4e89894d55796970f601c368c14f251b80bd38cd3b
-size 6391
+oid sha256:b7d791fed505e7d3ed82e828e487afd7c94a54a2064ef37317ce35a78c66461a
+size 6745