End of training

README.md

@@ -1,7 +1,7 @@
 ---
 library_name: transformers
 license: mit
-base_model: facebook/bart-large-xsum
+base_model: SumanNazir/highlight_summary_model_trained_on_reduced_data
 tags:
 - generated_from_trainer
 model-index:
@@ -14,9 +14,19 @@ should probably proofread and complete it, then remove this comment. -->
 
 # highlight_summary_model_trained_on_reduced_data
 
-This model is a fine-tuned version of [facebook/bart-large-xsum](https://huggingface.co/facebook/bart-large-xsum) on an unknown dataset.
+This model is a fine-tuned version of [SumanNazir/highlight_summary_model_trained_on_reduced_data](https://huggingface.co/SumanNazir/highlight_summary_model_trained_on_reduced_data) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.7400
+- eval_loss: 4.4296
+- eval_rouge1: 0.4082
+- eval_rouge2: 0.1596
+- eval_rougeL: 0.3068
+- eval_rougeLsum: 0.3073
+- eval_generated_length: 36.5303
+- eval_runtime: 99.8804
+- eval_samples_per_second: 1.322
+- eval_steps_per_second: 0.661
+- epoch: 1.0
+- step: 263
 
 ## Model description
 
@@ -36,25 +46,17 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
-- train_batch_size: 8
-- eval_batch_size: 8
+- train_batch_size: 2
+- eval_batch_size: 2
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - num_epochs: 2
 - mixed_precision_training: Native AMP
 
-### Training results
-
-| Training Loss | Epoch | Step | Validation Loss |
-|:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 274  | 1.7445          |
-| 1.6984        | 2.0   | 548  | 1.7400          |
-
-
 ### Framework versions
 
-- Transformers 4.44.2
+- Transformers 4.45.1
 - Pytorch 2.4.1+cu121
 - Datasets 3.0.1
-- Tokenizers 0.19.1
+- Tokenizers 0.20.0

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "facebook/bart-large-xsum",
+  "_name_or_path": "SumanNazir/highlight_summary_model_trained_on_reduced_data",
   "_num_labels": 3,
   "activation_dropout": 0.0,
   "activation_function": "gelu",
@@ -19,7 +19,7 @@
   "decoder_layers": 12,
   "decoder_start_token_id": 2,
   "dropout": 0.1,
-  "early_stopping": true,
+  "early_stopping": null,
   "encoder_attention_heads": 16,
   "encoder_ffn_dim": 4096,
   "encoder_layerdrop": 0.0,
@@ -42,14 +42,14 @@
     "LABEL_1": 1,
     "LABEL_2": 2
   },
-  "max_length": 62,
+  "max_length": null,
   "max_position_embeddings": 1024,
-  "min_length": 11,
+  "min_length": null,
   "model_type": "bart",
-  "no_repeat_ngram_size": 3,
+  "no_repeat_ngram_size": null,
   "normalize_before": false,
   "normalize_embedding": true,
-  "num_beams": 6,
+  "num_beams": null,
   "num_hidden_layers": 12,
   "output_past": true,
   "pad_token_id": 1,
@@ -61,7 +61,7 @@
   "student_encoder_layers": null,
   "task_specific_params": {},
   "torch_dtype": "float32",
-  "transformers_version": "4.44.2",
+  "transformers_version": "4.45.1",
   "use_cache": true,
   "vocab_size": 50264
 }

generation_config.json

@@ -1,5 +1,4 @@
 {
-  "_from_model_config": true,
   "bos_token_id": 0,
   "decoder_start_token_id": 2,
   "early_stopping": true,
@@ -10,5 +9,5 @@
   "no_repeat_ngram_size": 3,
   "num_beams": 6,
   "pad_token_id": 1,
-  "transformers_version": "4.44.2"
+  "transformers_version": "4.45.1"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d66c059ea26aa5cd256e556385200e2a5dad918635b21d5534409b1141106f37
+oid sha256:f92a3de772866d1caec2c83b213871c9a22e2a0410fcf9136c0c36bd08c1ac84
 size 1625422896

runs/Sep30_19-38-35_34c3efdfaef4/events.out.tfevents.1727725116.34c3efdfaef4.11954.10

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

runs/Sep30_19-39-15_34c3efdfaef4/events.out.tfevents.1727725156.34c3efdfaef4.11954.11

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

runs/Sep30_19-39-57_34c3efdfaef4/events.out.tfevents.1727725198.34c3efdfaef4.11954.12

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

runs/Sep30_19-40-47_34c3efdfaef4/events.out.tfevents.1727725248.34c3efdfaef4.11954.13

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

runs/Sep30_19-46-20_34c3efdfaef4/events.out.tfevents.1727725581.34c3efdfaef4.11954.14

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

special_tokens_map.json

@@ -1,7 +1,25 @@
 {
-  "bos_token": "<s>",
-  "cls_token": "<s>",
-  "eos_token": "</s>",
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "cls_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
   "mask_token": {
     "content": "<mask>",
     "lstrip": true,
@@ -9,7 +27,25 @@
     "rstrip": false,
     "single_word": false
   },
-  "pad_token": "<pad>",
-  "sep_token": "</s>",
-  "unk_token": "<unk>"
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "sep_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  }
 }

tokenizer_config.json

@@ -48,10 +48,14 @@
   "eos_token": "</s>",
   "errors": "replace",
   "mask_token": "<mask>",
+  "max_length": 128,
   "model_max_length": 1024,
   "pad_token": "<pad>",
   "sep_token": "</s>",
+  "stride": 0,
   "tokenizer_class": "BartTokenizer",
   "trim_offsets": true,
+  "truncation_side": "right",
+  "truncation_strategy": "longest_first",
   "unk_token": "<unk>"
 }

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6035351f6f9d1fa4425975b1fe2fc2c505ee88cbad734c53c485bbb25c9e4e79
-size 5368
+oid sha256:bee5d249d7dbc780063c25d8ce1821d4b28bb73be69b152b8b78e0324e67a39c
+size 5432