Training complete

README.md

@@ -0,0 +1,67 @@
+---
+library_name: transformers
+license: apache-2.0
+base_model: google/mt5-small
+tags:
+- summarization
+- generated_from_trainer
+metrics:
+- rouge
+model-index:
+- name: mt5-small-finetuned-amazon-en-es
+  results: []
+---
+
+<!-- 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. -->
+
+# mt5-small-finetuned-amazon-en-es
+
+This model is a fine-tuned version of [google/mt5-small](https://huggingface.co/google/mt5-small) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 3.2668
+- Rouge1: 14.1234
+- Rouge2: 5.8438
+- Rougel: 13.7241
+- Rougelsum: 13.6458
+
+## 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: 16
+- eval_batch_size: 16
+- seed: 42
+- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
+- lr_scheduler_type: linear
+- num_epochs: 3
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Rouge1  | Rouge2 | Rougel  | Rougelsum |
+|:-------------:|:-----:|:----:|:---------------:|:-------:|:------:|:-------:|:---------:|
+| 5.0289        | 1.0   | 605  | 3.4412          | 11.4133 | 3.91   | 11.157  | 11.1867   |
+| 4.252         | 2.0   | 1210 | 3.2869          | 13.9412 | 6.0411 | 13.6726 | 13.5899   |
+| 4.0688        | 3.0   | 1815 | 3.2668          | 14.1234 | 5.8438 | 13.7241 | 13.6458   |
+
+
+### Framework versions
+
+- Transformers 4.46.2
+- Pytorch 2.5.1+cu121
+- Datasets 3.1.0
+- Tokenizers 0.20.3

generation_config.json

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

runs/Dec04_04-34-44_38ab3c76815c/events.out.tfevents.1733286912.38ab3c76815c.303.1

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e70f5c735fce111ba42b545214c93aec5f6ac22bace339f706c4d8b35061da94
-size 8237
+oid sha256:4355597f58942e860ac18cee357425f73810e71ba24ce71d1b6715c376e03936
+size 9065

runs/Dec04_04-34-44_38ab3c76815c/events.out.tfevents.1733288157.38ab3c76815c.303.2

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