Training complete

README.md

@@ -0,0 +1,80 @@
+---
+license: mit
+base_model: facebook/bart-large-cnn
+tags:
+- summarization
+- generated_from_trainer
+datasets:
+- tldr_news
+metrics:
+- rouge
+model-index:
+- name: summary_model
+  results:
+  - task:
+      name: Sequence-to-sequence Language Modeling
+      type: text2text-generation
+    dataset:
+      name: tldr_news
+      type: tldr_news
+      config: all
+      split: test
+      args: all
+    metrics:
+    - name: Rouge1
+      type: rouge
+      value: 0.21590240799799404
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# summary_model
+
+This model is a fine-tuned version of [facebook/bart-large-cnn](https://huggingface.co/facebook/bart-large-cnn) on the tldr_news dataset.
+It achieves the following results on the evaluation set:
+- Loss: 2.9573
+- Rouge1: 0.2159
+- Rouge2: 0.0831
+- Rougel: 0.1829
+- Rougelsum: 0.1869
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 5.6e-05
+- 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
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
+|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|
+| 0.5871        | 1.0   | 63   | 2.7134          | 0.2176 | 0.0872 | 0.1881 | 0.1951    |
+| 0.4422        | 2.0   | 126  | 2.9573          | 0.2159 | 0.0831 | 0.1829 | 0.1869    |
+
+
+### Framework versions
+
+- Transformers 4.35.2
+- Pytorch 2.1.0+cu121
+- Datasets 2.16.1
+- Tokenizers 0.15.0

generation_config.json

@@ -0,0 +1,15 @@
+{
+  "bos_token_id": 0,
+  "decoder_start_token_id": 2,
+  "early_stopping": true,
+  "eos_token_id": 2,
+  "forced_bos_token_id": 0,
+  "forced_eos_token_id": 2,
+  "length_penalty": 2.0,
+  "max_length": 142,
+  "min_length": 56,
+  "no_repeat_ngram_size": 3,
+  "num_beams": 4,
+  "pad_token_id": 1,
+  "transformers_version": "4.35.2"
+}

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7b315f75e6d43200b841b20140cdce09276e3b23d3d802a6bf897e84fc953793
+oid sha256:095f033424729db36ba99b28ff57c639790bafcdff760c09f55c7cc02f95b3e4
 size 1625422896

runs/Jan24_11-43-50_95fda9807fbe/events.out.tfevents.1706096662.95fda9807fbe.2753.0

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

runs/Jan24_11-43-50_95fda9807fbe/events.out.tfevents.1706099295.95fda9807fbe.2753.1

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

runs/Jan24_13-27-30_95fda9807fbe/events.out.tfevents.1706102867.95fda9807fbe.2753.2

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

runs/Jan24_13-27-30_95fda9807fbe/events.out.tfevents.1706103387.95fda9807fbe.2753.3

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

runs/Jan24_13-27-30_95fda9807fbe/events.out.tfevents.1706103674.95fda9807fbe.2753.4

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

runs/Jan24_13-27-30_95fda9807fbe/events.out.tfevents.1706104371.95fda9807fbe.2753.5

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:24fca1ed5db0518ce17f625c9c2f2955a495013d5395f3f788bfdc784d9878ea
+oid sha256:eec0db9fe404c68f89bdb46358fd624f10805cc8fbf6de0fa0a6964aa4071188
 size 4728