Edit model card

t5-small-finetuned-topics-customdata

This model is a fine-tuned version of t5-small on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 2.6454
  • Rouge1: 35.5152
  • Rouge2: 26.0061
  • Rougel: 34.6245
  • Rougelsum: 34.6301
  • Gen Len: 12.131

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: 2e-05
  • train_batch_size: 2
  • eval_batch_size: 2
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 3

Training results

Training Loss Epoch Step Validation Loss Rouge1 Rouge2 Rougel Rougelsum Gen Len
2.6449 1.0 1000 2.6940 34.7463 25.5926 33.9278 33.9533 12.456
2.4802 2.0 2000 2.6530 35.3548 25.9763 34.4508 34.4747 12.329
2.4914 3.0 3000 2.6454 35.5152 26.0061 34.6245 34.6301 12.131

Framework versions

  • Transformers 4.28.1
  • Pytorch 2.0.0+cu118
  • Datasets 2.11.0
  • Tokenizers 0.13.3
Downloads last month
7