End of training

README.md

@@ -3,6 +3,8 @@ license: apache-2.0
 base_model: Falconsai/text_summarization
 tags:
 - generated_from_trainer
+metrics:
+- rouge
 model-index:
 - name: text_summarization_finetuned2
   results: []
@@ -15,17 +17,12 @@ 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 the None dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 0.2328
-- eval_rouge1: 0.0876
-- eval_rouge2: 0.0826
-- eval_rougeL: 0.0876
-- eval_rougeLsum: 0.0876
-- eval_gen_len: 19.0
-- eval_runtime: 1482.5408
-- eval_samples_per_second: 6.262
-- eval_steps_per_second: 6.262
-- epoch: 2.0
-- step: 74258
+- Loss: 0.3126
+- Rouge1: 0.0675
+- Rouge2: 0.0578
+- Rougel: 0.0674
+- Rougelsum: 0.0674
+- Gen Len: 19.0
 
 ## Model description
 
@@ -45,14 +42,23 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
-- train_batch_size: 1
-- eval_batch_size: 1
+- train_batch_size: 2
+- eval_batch_size: 2
 - seed: 42
 - distributed_type: multi-GPU
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - num_epochs: 4
-- mixed_precision_training: Native AMP
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
+|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
+| 0.4139        | 1.0   | 2000 | 0.3411          | 0.0632 | 0.0524 | 0.0632 | 0.0632    | 19.0    |
+| 0.3635        | 2.0   | 4000 | 0.3215          | 0.0658 | 0.0557 | 0.0658 | 0.0658    | 19.0    |
+| 0.348         | 3.0   | 6000 | 0.3146          | 0.0668 | 0.0571 | 0.0668 | 0.0668    | 19.0    |
+| 0.3445        | 4.0   | 8000 | 0.3126          | 0.0675 | 0.0578 | 0.0674 | 0.0674    | 19.0    |
+
 
 ### Framework versions
 

model.safetensors

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

runs/Apr25_15-47-58_0423-173034-yjob166g-10-228-67-11/events.out.tfevents.1714060080.0423-173034-yjob166g-10-228-67-11.2159.1

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

runs/Apr25_15-49-41_0423-173034-yjob166g-10-228-67-11/events.out.tfevents.1714060183.0423-173034-yjob166g-10-228-67-11.2159.2

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

runs/Apr25_17-08-17_0423-173034-yjob166g-10-228-67-32/events.out.tfevents.1714064903.0423-173034-yjob166g-10-228-67-32.2180.0

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

runs/Apr25_17-38-50_0423-173034-yjob166g-10-228-67-32/events.out.tfevents.1714066732.0423-173034-yjob166g-10-228-67-32.2180.1

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:02f9c42980689ffab05eba54c6413b4fdde159348e4371c2fd913c1fa7c3fba8
-size 4984
+oid sha256:bca2ad486a3b6c0e3e0c8b2acb6cdee38c5a40d41d749079a2da5812b3197667
+size 4920