End of training

README.md

@@ -5,10 +5,24 @@ tags:
 - generated_from_trainer
 datasets:
 - xsum
+metrics:
+- rouge
 model-index:
 - name: t5-small-finetuned-xsum
-  results: []
-pipeline_tag: summarization
+  results:
+  - task:
+      name: Sequence-to-sequence Language Modeling
+      type: text2text-generation
+    dataset:
+      name: xsum
+      type: xsum
+      config: default
+      split: validation
+      args: default
+    metrics:
+    - name: Rouge1
+      type: rouge
+      value: 28.2959
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -17,6 +31,13 @@ should probably proofread and complete it, then remove this comment. -->
 # t5-small-finetuned-xsum
 
 This model is a fine-tuned version of [google-t5/t5-small](https://huggingface.co/google-t5/t5-small) on the xsum dataset.
+It achieves the following results on the evaluation set:
+- Loss: 2.4781
+- Rouge1: 28.2959
+- Rouge2: 7.7364
+- Rougel: 22.2437
+- Rougelsum: 22.2447
+- Gen Len: 18.8252
 
 ## Model description
 
@@ -44,9 +65,16 @@ The following hyperparameters were used during training:
 - num_epochs: 1
 - mixed_precision_training: Native AMP
 
+### Training results
+
+| Training Loss | Epoch | Step  | Validation Loss | Rouge1  | Rouge2 | Rougel  | Rougelsum | Gen Len |
+|:-------------:|:-----:|:-----:|:---------------:|:-------:|:------:|:-------:|:---------:|:-------:|
+| 2.7071        | 1.0   | 12753 | 2.4781          | 28.2959 | 7.7364 | 22.2437 | 22.2447   | 18.8252 |
+
+
 ### Framework versions
 
 - Transformers 4.41.2
 - Pytorch 2.1.2
 - Datasets 2.19.2
-- Tokenizers 0.19.1
+- Tokenizers 0.19.1

generation_config.json

@@ -1,5 +1,4 @@
 {
-  "_from_model_config": true,
   "decoder_start_token_id": 0,
   "eos_token_id": 1,
   "pad_token_id": 0,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:98f960fbdfffd58fd750797b5640f4306dff2b7217a562a396f6875f1750e37a
+oid sha256:80329fd22851f03ceee4c2aab3399a34e91ebce5eea98cbf064352d515610a62
 size 242041896

runs/Jun20_04-29-11_d39f9482762b/events.out.tfevents.1718857752.d39f9482762b.24.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b982e7bbc66c99fe307c8e53e1df22f4560616ea7262d4011b5893261ec0935d
-size 11109
+oid sha256:989189c780c2c5a076b02781f4b11f2cccc6937b758bfb17fd93db7faf1d6fa6
+size 11988