Training complete

README.md

@@ -0,0 +1,68 @@
+---
+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.0961
+- Rouge1: 18.0893
+- Rouge2: 8.8114
+- Rougel: 17.2241
+- Rougelsum: 17.3175
+
+## 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: 5
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Rouge1  | Rouge2 | Rougel  | Rougelsum |
+|:-------------:|:-----:|:----:|:---------------:|:-------:|:------:|:-------:|:---------:|
+| 6.8796        | 1.0   | 1209 | 3.3045          | 15.001  | 6.3129 | 14.4408 | 14.521    |
+| 3.9359        | 2.0   | 2418 | 3.1826          | 15.9318 | 7.5821 | 15.1938 | 15.235    |
+| 3.6446        | 3.0   | 3627 | 3.1361          | 17.4174 | 8.7074 | 16.6003 | 16.7056   |
+| 3.5156        | 4.0   | 4836 | 3.1162          | 17.6097 | 8.5954 | 16.8641 | 16.9235   |
+| 3.4365        | 5.0   | 6045 | 3.0961          | 18.0893 | 8.8114 | 17.2241 | 17.3175   |
+
+
+### Framework versions
+
+- Transformers 4.46.2
+- Pytorch 2.5.1+cu121
+- 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:8f38dff9c3b34420f88f15f4ccfced2b33597b726a75156fa4df716e57cd8deb
+oid sha256:d89b0134b01a64628caa2ab008d28b35fea954efa4ca16eadab55f3176e6f92e
 size 1200729512

runs/Dec03_20-59-19_f3a76824ace0/events.out.tfevents.1733259607.f3a76824ace0.925.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c068d1e9809141f94eaa00272e3a176624477928b2b2765b7acd9d238d9030e5
-size 8129
+oid sha256:7ae1f012c9aab4dffa070d72f167c1c35f38e62bbb7896d94e29a102ab9aee6d
+size 9168

runs/Dec03_20-59-19_f3a76824ace0/events.out.tfevents.1733262600.f3a76824ace0.925.1

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