Edit model card

t5-small-finetuned-tf-xsum

This model was trained from scratch on xsum dataset. It achieves the following results on the evaluation set:

  • Train Loss: 2.3494
  • Validation Loss: 2.1933
  • Train Rouge1: 32.0241
  • Train Rouge2: 10.1025
  • Train Rougel: 25.8834
  • Train Rougelsum: 25.9662
  • Train Gen Len: 18.69
  • Epoch: 8

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:

  • optimizer: {'name': 'AdamWeightDecay', 'learning_rate': 2e-05, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False, 'weight_decay_rate': 0.01}
  • training_precision: float32

Training results

Train Loss Validation Loss Train Rouge1 Train Rouge2 Train Rougel Train Rougelsum Train Gen Len Epoch
2.7197 2.4028 29.6376 8.8596 22.8598 22.8401 18.82 1
2.5822 2.3407 30.6849 9.3100 23.8971 23.9096 18.745 2
2.5174 2.2979 32.3706 11.5463 26.4253 26.3525 18.75 3
2.4711 2.2703 32.2768 11.0460 26.2472 26.1540 18.825 4
2.4305 2.2432 29.3935 8.3337 22.2859 22.3557 18.65 5
2.3994 2.2237 31.0993 8.7932 23.6971 23.7702 18.815 6
2.3732 2.2071 31.4819 10.0677 25.1846 25.2829 18.675 7
2.3494 2.1933 32.0241 10.1025 25.8834 25.9662 18.69 8

Framework versions

  • Transformers 4.21.1
  • TensorFlow 2.8.2
  • Datasets 2.4.0
  • Tokenizers 0.12.1
Downloads last month
9
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.