mt5-small-finetuned-amazon-en-de

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: 2.7569
• Rouge1: 18.0555
• Rouge2: 9.7451
• Rougel: 17.6297
• Rougelsum: 17.5489

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: 5e-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: 4

Training results

Training Loss	Epoch	Step	Validation Loss	Rouge1	Rouge2	Rougel	Rougelsum
2.7527	1.0	3665	2.8879	18.3899	10.2549	17.9226	17.8373
3.1875	2.0	7330	2.7905	17.6365	9.271	17.1788	17.0681
3.0707	3.0	10995	2.7601	17.8343	9.5332	17.4487	17.3742
3.0107	4.0	14660	2.7569	18.0555	9.7451	17.6297	17.5489

Framework versions

• Transformers 4.26.1
• Pytorch 1.13.1+cu116
• Datasets 2.10.1
• Tokenizers 0.13.2