mt5-small-finetuned-amazon-en-es

This model is a fine-tuned version of google/mt5-small on the None dataset. It achieves the following results on the evaluation set:

• Loss: 3.0340
• Rouge1: 17.3066
• Rouge2: 8.5372
• Rougel: 16.9577
• Rougelsum: 17.1267

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: Adam with betas=(0.9,0.999) and epsilon=1e-08
• lr_scheduler_type: linear
• num_epochs: 8

Training results

Training Loss	Epoch	Step	Validation Loss	Rouge1	Rouge2	Rougel	Rougelsum
7.0197	1.0	1209	3.3037	13.7225	5.4609	13.1771	13.2052
3.9145	2.0	2418	3.1418	15.6039	7.5306	14.9366	14.865
3.5987	3.0	3627	3.0970	17.425	8.6602	16.9049	17.0042
3.4274	4.0	4836	3.0672	16.7739	8.0707	16.2041	16.2127
3.3241	5.0	6045	3.0648	16.6489	8.2121	16.3527	16.4147
3.2468	6.0	7254	3.0444	17.3052	8.6923	16.9398	17.0233
3.2116	7.0	8463	3.0370	17.563	8.7613	17.1755	17.3348
3.1821	8.0	9672	3.0340	17.3066	8.5372	16.9577	17.1267

Framework versions

• Transformers 4.30.2
• Pytorch 2.0.1+cu118
• Datasets 2.13.1
• Tokenizers 0.13.3