Edit model card

flan-t5-base-samsum

This model is a fine-tuned version of google/flan-t5-base on the samsum dataset. It achieves the following results on the evaluation set:

  • Loss: 1.3721
  • Rouge1: 47.698
  • Rouge2: 23.8078
  • Rougel: 40.1138
  • Rougelsum: 43.7749
  • Gen Len: 17.2759

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

Training results

Training Loss Epoch Step Validation Loss Rouge1 Rouge2 Rougel Rougelsum Gen Len
1.4403 1.0 1842 1.3822 47.3182 23.8486 39.7145 43.5756 17.0256
1.3572 2.0 3684 1.3747 47.5891 23.6341 39.7983 43.6862 17.4347
1.2822 3.0 5526 1.3721 47.698 23.8078 40.1138 43.7749 17.2759
1.2375 4.0 7368 1.3764 47.7671 24.1413 40.1597 43.9313 17.2943
1.1935 5.0 9210 1.3781 47.626 23.7564 39.844 43.7166 17.3077

Framework versions

  • Transformers 4.27.3
  • Pytorch 1.13.1+cu116
  • Datasets 2.10.1
  • Tokenizers 0.13.2
Downloads last month
12

Dataset used to train szilard/flan-t5-base-samsum

Evaluation results