Edit model card

Prototypeu/bart-base-finetuned-tldrhq-cnn-dailymail

This model is a fine-tuned version of Prototypeu/bart-base-finetuned-xsum on an unknown dataset. It achieves the following results on the evaluation set:

  • Train Loss: 1.5049
  • Train Logits Loss: 1.5049
  • Train Rouge1: 28.1795
  • Train Rouge2: 14.0392
  • Train Rougel: 23.7617
  • Train Rougelsum: 26.5583
  • Train Gen Len: 19.0
  • Epoch: 4

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': {'class_name': 'WarmUp', 'config': {'initial_learning_rate': 3e-05, 'decay_schedule_fn': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 3e-05, 'decay_steps': 255113, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'passive_serialization': True}, 'warmup_steps': 32000, 'power': 1.0, 'name': None}}, 'decay': 0.0, 'beta_1': 0.98, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
  • training_precision: float32

Training results

Train Loss Train Logits Loss Train Rouge1 Train Rouge2 Train Rougel Train Rougelsum Train Gen Len Epoch
2.9074 2.9074 26.9164 12.6984 22.4321 25.2287 19.0 0
1.9368 1.9368 28.0165 13.8906 23.4187 26.3779 19.0 1
1.7246 1.7246 27.6022 13.5255 23.2301 25.9923 19.0 2
1.5945 1.5945 28.0347 13.7045 23.4851 26.3488 19.0 3
1.5049 1.5049 28.1795 14.0392 23.7617 26.5583 19.0 4

Framework versions

  • Transformers 4.17.0
  • TensorFlow 2.6.0
  • Datasets 2.0.0
  • Tokenizers 0.11.6
Downloads last month
2
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.