models / README.md
Hendrik-a's picture
End of training
b68c003
metadata
base_model: google/pegasus-pubmed
tags:
  - generated_from_trainer
metrics:
  - rouge
model-index:
  - name: models
    results: []

models

This model is a fine-tuned version of google/pegasus-pubmed on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 2.8844
  • Rouge1: 36.8587
  • Rouge2: 12.7373
  • Rougel: 23.9623
  • Rougelsum: 31.0333
  • Gen Len: 106.2

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: 0.0001
  • train_batch_size: 4
  • eval_batch_size: 4
  • seed: 100
  • optimizer: Adafactor
  • lr_scheduler_type: linear
  • num_epochs: 1.0

Training results

Training Loss Epoch Step Validation Loss Rouge1 Rouge2 Rougel Rougelsum Gen Len
No log 1.0 100 2.8844 36.5048 15.2772 23.4469 31.6065 181.9

Framework versions

  • Transformers 4.33.2
  • Pytorch 2.0.1+cu118
  • Datasets 2.14.5
  • Tokenizers 0.13.3