<!-- This model card has been generated automatically according to the information Keras had access to. You should probably proofread and complete it, then remove this comment. -->
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
