flan-t5-small-samsum
This model is a fine-tuned version of google/flan-t5-small on an unknown dataset. It achieves the following results on the evaluation set:
- Loss: 1.6231
- Rouge1: 46.1671
- Rouge2: 21.1856
- Rougel: 38.1757
- Rougelsum: 42.2628
- Gen Len: 23.9548
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: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: linear
- num_epochs: 5
Training results
| Training Loss | Epoch | Step | Gen Len | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
|---|---|---|---|---|---|---|---|---|
| 1.4770 | 1.0 | 1842 | 24.3431 | 1.6961 | 44.2122 | 19.8283 | 36.4978 | 40.3151 |
| 1.6611 | 2.0 | 3684 | 23.1294 | 1.6280 | 45.5456 | 20.6914 | 37.732 | 41.7582 |
| 1.6080 | 3.0 | 5526 | 23.5592 | 1.6243 | 45.9714 | 21.1067 | 38.1001 | 42.0213 |
| 1.6034 | 4.0 | 7368 | 23.9548 | 1.6231 | 46.1671 | 21.1856 | 38.1757 | 42.2628 |
| 1.5841 | 5.0 | 9210 | 23.7204 | 1.6231 | 46.2543 | 21.2934 | 38.2606 | 42.2705 |
Framework versions
- Transformers 5.5.4
- Pytorch 2.10.0+cu128
- Datasets 4.8.4
- Tokenizers 0.22.2