led-large-16384-BASE-Attention3ep

This model is a fine-tuned version of allenai/led-large-16384 on the None dataset. It achieves the following results on the evaluation set:

• Loss: 2.0410
• Rouge1: 42.1053
• Rouge2: 15.7014
• Rougel: 19.2555
• Rougelsum: 39.5379
• Gen Len: 1.0

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
• gradient_accumulation_steps: 2
• total_train_batch_size: 16
• optimizer: Use OptimizerNames.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: 3
• mixed_precision_training: Native AMP

Training results

Training Loss	Epoch	Step	Validation Loss	Rouge1	Rouge2	Rougel	Rougelsum	Gen Len
2.897	1.0	1132	2.1394	40.613	13.5723	20.9451	39.0805	1.0
2.6659	2.0	2264	2.0424	42.7481	15.5612	21.1196	41.9847	1.0
2.4834	2.9978	3393	2.0410	42.1053	15.7014	19.2555	39.5379	1.0

Framework versions

• Transformers 4.48.3
• Pytorch 2.11.0+cu128
• Datasets 4.8.5
• Tokenizers 0.21.4