Training in progress, step 500

README.md

@@ -1,7 +1,7 @@
 ---
 library_name: transformers
 license: mit
-base_model: SumanNazir/highlight_summary_model_trained_on_reduced_data
+base_model: facebook/bart-large-xsum
 tags:
 - generated_from_trainer
 model-index:
@@ -14,19 +14,9 @@ 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 [SumanNazir/highlight_summary_model_trained_on_reduced_data](https://huggingface.co/SumanNazir/highlight_summary_model_trained_on_reduced_data) on an unknown dataset.
+This model is a fine-tuned version of [facebook/bart-large-xsum](https://huggingface.co/facebook/bart-large-xsum) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- 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
+- Loss: 1.7400
 
 ## Model description
 
@@ -46,17 +36,25 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
-- train_batch_size: 2
-- eval_batch_size: 2
+- train_batch_size: 8
+- eval_batch_size: 8
 - 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.45.1
+- Transformers 4.44.2
 - Pytorch 2.4.1+cu121
 - Datasets 3.0.1
-- Tokenizers 0.20.0
+- Tokenizers 0.19.1

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "SumanNazir/highlight_summary_model_trained_on_reduced_data",
+  "_name_or_path": "facebook/bart-large-xsum",
   "_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": null,
+  "early_stopping": true,
   "encoder_attention_heads": 16,
   "encoder_ffn_dim": 4096,
   "encoder_layerdrop": 0.0,
@@ -42,14 +42,14 @@
     "LABEL_1": 1,
     "LABEL_2": 2
   },
-  "max_length": null,
+  "max_length": 62,
   "max_position_embeddings": 1024,
-  "min_length": null,
+  "min_length": 11,
   "model_type": "bart",
-  "no_repeat_ngram_size": null,
+  "no_repeat_ngram_size": 3,
   "normalize_before": false,
   "normalize_embedding": true,
-  "num_beams": null,
+  "num_beams": 6,
   "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.45.1",
+  "transformers_version": "4.44.2",
   "use_cache": true,
   "vocab_size": 50264
 }

generation_config.json

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

model.safetensors

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

runs/Sep30_19-58-10_13959885f477/events.out.tfevents.1727726291.13959885f477.218.4

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

special_tokens_map.json

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

tokenizer_config.json

@@ -48,14 +48,10 @@
   "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:040a829b06db162606556c2f6239102a308d4530eaa19d03f1bf19f978ba81dd
-size 5432
+oid sha256:566c9ecb6f250c606f2f481a33a57aaa068c75a76e5bf7bec15d879ae45b7437
+size 5368