mT5_small

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: 0.3417
• Bleu Score: 47.0526
• Gen Len: 16.8315

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: 0.0001
• 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: 4

Training results

Training Loss	Epoch	Step	Validation Loss	Bleu Score	Gen Len
2.798	1.0	838	0.5495	41.8683	16.7766
0.7216	2.0	1676	0.4311	44.9002	16.8148
0.5551	3.0	2514	0.3565	46.5247	16.816
0.4951	4.0	3352	0.3417	47.0526	16.8315

Framework versions

• Transformers 4.46.2
• Pytorch 2.5.0+cu121
• Datasets 3.1.0
• Tokenizers 0.20.3