Training complete

README.md

@@ -0,0 +1,72 @@
+---
+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.0303
+- Rouge1: 16.5779
+- Rouge2: 7.6628
+- Rougel: 16.1432
+- Rougelsum: 16.0108
+
+## 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: 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: 8
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Rouge1  | Rouge2 | Rougel  | Rougelsum |
+|:-------------:|:-----:|:----:|:---------------:|:-------:|:------:|:-------:|:---------:|
+| 6.9675        | 1.0   | 1209 | 3.2986          | 15.3253 | 6.9003 | 14.889  | 14.7934   |
+| 3.8997        | 2.0   | 2418 | 3.1665          | 16.4209 | 7.6246 | 15.8049 | 15.6988   |
+| 3.5826        | 3.0   | 3627 | 3.1106          | 17.2164 | 8.4302 | 16.5395 | 16.3992   |
+| 3.421         | 4.0   | 4836 | 3.0963          | 17.3394 | 8.7889 | 16.8507 | 16.7652   |
+| 3.3089        | 5.0   | 6045 | 3.0490          | 16.7239 | 7.5913 | 16.2355 | 16.1438   |
+| 3.2437        | 6.0   | 7254 | 3.0401          | 16.6428 | 7.9473 | 16.0411 | 15.9346   |
+| 3.2133        | 7.0   | 8463 | 3.0292          | 16.3094 | 7.634  | 15.9688 | 15.8809   |
+| 3.1851        | 8.0   | 9672 | 3.0303          | 16.5779 | 7.6628 | 16.1432 | 16.0108   |
+
+
+### 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"
+}

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:af5ff943710fd83f3087bc49a57f18d2f4bf85e883b1ffe0bcaa08606512d6d9
+oid sha256:b7ea2dd42c79fcfe86a56691a782c296c60e65eaab3efe3c7ab57cb2067add32
 size 1200729512

runs/Dec03_18-33-07_22356dbe70a0/events.out.tfevents.1733250806.22356dbe70a0.429.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f3d19215926c5b566d6a765acc2da7657c0133d49ef9cf4b981c47f2da060eeb
-size 10184
+oid sha256:8273bc64f2cbc2ec3ca7423dadef9019de31495c99455f62fa5bb94e50541aa0
+size 11223

runs/Dec03_18-33-07_22356dbe70a0/events.out.tfevents.1733254538.22356dbe70a0.429.1

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