Model save

README.md

@@ -1,12 +1,82 @@
 ---
 license: apache-2.0
+base_model: t5-small
+tags:
+- generated_from_trainer
 datasets:
 - xsum
-language:
-- en
 metrics:
 - rouge
-tags:
-- NLP
-- summarization
----
+model-index:
+- name: T5-XSum-base
+  results:
+  - task:
+      name: Sequence-to-sequence Language Modeling
+      type: text2text-generation
+    dataset:
+      name: xsum
+      type: xsum
+      config: default
+      split: train
+      args: default
+    metrics:
+    - name: Rouge1
+      type: rouge
+      value: 0.273
+---
+
+<!-- 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. -->
+
+# T5-XSum-base
+
+This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on the xsum dataset.
+It achieves the following results on the evaluation set:
+- Loss: 2.5491
+- Rouge1: 0.273
+- Rouge2: 0.0711
+- Rougel: 0.2134
+- Rougelsum: 0.2134
+- Gen Len: 18.8194
+
+## 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: 2e-05
+- train_batch_size: 16
+- eval_batch_size: 16
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- num_epochs: 3
+- mixed_precision_training: Native AMP
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
+|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
+| 2.8234        | 1.0   | 2041 | 2.5916          | 0.2623 | 0.0647 | 0.2043 | 0.2044    | 18.8152 |
+| 2.7742        | 2.0   | 4082 | 2.5577          | 0.2707 | 0.0702 | 0.2118 | 0.2117    | 18.8212 |
+| 2.7482        | 3.0   | 6123 | 2.5491          | 0.273  | 0.0711 | 0.2134 | 0.2134    | 18.8194 |
+
+
+### Framework versions
+
+- Transformers 4.35.0
+- Pytorch 1.12.0+cu116
+- Datasets 2.14.6
+- Tokenizers 0.14.1

generation_config.json

@@ -0,0 +1,6 @@
+{
+  "decoder_start_token_id": 0,
+  "eos_token_id": 1,
+  "pad_token_id": 0,
+  "transformers_version": "4.35.0"
+}

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8de0943e169f7f3b7da9eed5f0601432709db42c2c1cd418e7ddc9086b03a44e
+oid sha256:4b6d5ebd1c16bdc5a3aa4d7e5bb5ed8d99a18c555dfea263b1af76f8e4147cff
 size 242041896

runs/Nov07_17-19-52_n22hf832wh/events.out.tfevents.1699377594.n22hf832wh.81.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7e28a9538d36d1fcdac3fd8dd5029adc72e2a181714800256a9ec7dc23a193fd
-size 8064
+oid sha256:459cc5d5617ac956e7d71013aa3a75e4014770b401e6ef78e4d260701214a55a
+size 8943