End of training

README.md

@@ -1,7 +1,7 @@
 ---
 library_name: transformers
 license: mit
-base_model: facebook/bart-large-cnn
+base_model: LearneratVnit/bart-summarizer
 tags:
 - generated_from_trainer
 metrics:
@@ -16,14 +16,14 @@ should probably proofread and complete it, then remove this comment. -->
 
 # bart-summarizer
 
-This model is a fine-tuned version of [facebook/bart-large-cnn](https://huggingface.co/facebook/bart-large-cnn) on the dataset available at https://zenodo.org/record/7152317#.Yz6mJ9JByC0.
+This model is a fine-tuned version of [LearneratVnit/bart-summarizer](https://huggingface.co/LearneratVnit/bart-summarizer) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 2.7172
-- Rouge1: 0.3278
-- Rouge2: 0.0669
-- Rougel: 0.2041
-- Rougelsum: 0.284
-- Gen Len: 133.6799
+- Loss: 2.3295
+- Rouge1: 0.33
+- Rouge2: 0.0688
+- Rougel: 0.205
+- Rougelsum: 0.2847
+- Gen Len: 129.5045
 
 ## Model description
 
@@ -50,17 +50,14 @@ The following hyperparameters were used during training:
 - total_train_batch_size: 8
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 4
+- num_epochs: 1
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch  | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len  |
 |:-------------:|:------:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:--------:|
-| 3.1498        | 0.9992 | 622  | 2.8687          | 0.3187 | 0.0635 | 0.1948 | 0.2765    | 141.2658 |
-| 2.7712        | 2.0    | 1245 | 2.7667          | 0.3195 | 0.0659 | 0.2006 | 0.2792    | 135.1609 |
-| 2.589         | 2.9992 | 1867 | 2.7298          | 0.3219 | 0.063  | 0.1962 | 0.2777    | 135.9729 |
-| 2.4744        | 3.9968 | 2488 | 2.7172          | 0.3278 | 0.0669 | 0.2041 | 0.284     | 133.6799 |
+| 2.4379        | 0.9992 | 622  | 2.3295          | 0.33   | 0.0688 | 0.205  | 0.2847    | 129.5045 |
 
 
 ### Framework versions