End of training

README.md

@@ -1,9 +1,11 @@
 ---
 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
+metrics:
+- rouge
 model-index:
 - name: highlight_summary_model_trained_on_reduced_data
   results: []
@@ -14,9 +16,14 @@ 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
+- Loss: 4.7393
+- Rouge1: 0.4167
+- Rouge2: 0.1684
+- Rougel: 0.3095
+- Rougelsum: 0.3088
+- Generated Length: 38.5758
 
 ## Model description
 
@@ -36,8 +43,8 @@ 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
@@ -46,15 +53,15 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss |
-|:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 274  | 1.7445          |
-| 1.6984        | 2.0   | 548  | 1.7400          |
+| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Generated Length |
+|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:----------------:|
+| No log        | 1.0   | 263  | 4.6745          | 0.4127 | 0.1629 | 0.3115 | 0.3119    | 38.0909          |
+| 0.0309        | 2.0   | 526  | 4.7393          | 0.4167 | 0.1684 | 0.3095 | 0.3088    | 38.5758          |
 
 
 ### 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:1f1902ad3dac6348954d33a6c0424a4bff3a35b6c07ae5a7c0dd5f979c3e9a32
+oid sha256:dae807a36ab2a73745aa6a8774bf07f05777b341764cb7cf949a7ec82109c834
 size 1625422896

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:8eda4ebfeadfc6e6c8c01390284062860c9e3a326b0825499ff6d00604fc3450
-size 5368
+oid sha256:040a829b06db162606556c2f6239102a308d4530eaa19d03f1bf19f978ba81dd
+size 5432