Training complete

README.md

@@ -1,6 +1,6 @@
 ---
 license: apache-2.0
-base_model: Falconsai/text_summarization
+base_model: google/flan-t5-base
 tags:
 - summarization
 - generated_from_trainer
@@ -23,7 +23,7 @@ model-index:
     metrics:
     - name: Rouge1
       type: rouge
-      value: 39.350569666957284
+      value: 44.34200782349507
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,11 +31,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 # notification-hub
 
-This model is a fine-tuned version of [Falconsai/text_summarization](https://huggingface.co/Falconsai/text_summarization) on the samsum dataset.
+This model is a fine-tuned version of [google/flan-t5-base](https://huggingface.co/google/flan-t5-base) on the samsum dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.8155
-- Rougel: 34.0230
-- Rouge1: 39.3506
+- Loss: 1.5633
+- Rougel: 38.3354
+- Rouge1: 44.3420
 
 ## Model description
 
@@ -66,15 +66,15 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Rougel  | Rouge1  |
 |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
-| 2.1248        | 1.0   | 2482 | 1.8711          | 32.6871 | 37.4214 |
-| 1.9801        | 2.0   | 4964 | 1.8399          | 33.6102 | 38.8617 |
-| 1.9584        | 3.0   | 7446 | 1.8219          | 33.6316 | 38.8673 |
-| 1.907         | 4.0   | 9928 | 1.8155          | 34.0230 | 39.3506 |
+| 1.6364        | 1.0   | 2482 | 1.5543          | 37.6025 | 43.5207 |
+| 1.4764        | 2.0   | 4964 | 1.5466          | 38.2764 | 44.0569 |
+| 1.3991        | 3.0   | 7446 | 1.5556          | 38.2109 | 43.9420 |
+| 1.3365        | 4.0   | 9928 | 1.5633          | 38.3354 | 44.3420 |
 
 
 ### Framework versions
 
 - Transformers 4.35.2
-- Pytorch 2.1.0+cu118
+- Pytorch 2.1.0+cu121
 - Datasets 2.15.0
 - Tokenizers 0.15.0

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ce881a71ac6dce4ead13ccacef5440a34584e7372ecd9cd9f366fea3c19e93bf
+oid sha256:d80245d7bde4cc4a8cb016fb5f24443903f74cd85c74d12a9a580f5fa13e1afb
 size 990345064

runs/Dec21_16-15-12_19caa3e25769/events.out.tfevents.1703175318.19caa3e25769.7799.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a79b368efd55f55f214d5d3844fddb4ff9382aa81461889f264741a5fdf4991c
-size 9342
+oid sha256:5d2e5629bf4255f4b09f2aa51c20aad132aa733ec285034cee935375f37dd87c
+size 10067

runs/Dec21_16-15-12_19caa3e25769/events.out.tfevents.1703179810.19caa3e25769.7799.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ae17ce735d715a524db5efb7fdd648c50b12f37adf39df5aa42cf36ea349e134
+size 459