jialicheng's picture
Upload folder using huggingface_hub
fd6ec32 verified
|
Raw History Blame Contribute Delete
3.29 kB
metadata
license: apache-2.0
base_model: google/t5-v1_1-base
tags:
  - generated_from_trainer
datasets:
  - samsum
metrics:
  - rouge
model-index:
  - name: samsum_42
    results:
      - task:
          name: Summarization
          type: summarization
        dataset:
          name: samsum
          type: samsum
        metrics:
          - name: Rouge1
            type: rouge
            value: 0

samsum_42

This model is a fine-tuned version of google/t5-v1_1-base on the samsum dataset. It achieves the following results on the evaluation set:

  • Loss: 89.9987
  • Rouge1: 0.0
  • Rouge2: 0.0
  • Rougel: 0.0
  • Rougelsum: 0.0
  • Gen Len: 127.0
  • Test Rougel: 0.0
  • Df Rougel: 0.0
  • Unlearn Overall Rougel: 0.5
  • Unlearn Time: 7305.7426

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: 32
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 5

Training results

Training Loss Epoch Step Validation Loss Rouge1 Rouge2 Rougel Rougelsum Gen Len Overall Rougel Unlearn Overall Rougel Time
No log 0.5 432 76.7971 0.0 0.0 0.0 0.0 127.0 0.5 0.5 -1
No log 1.0 864 82.2092 0.0 0.0 0.0 0.0 127.0 0.5 0.5 -1
0.0 1.5 1296 84.4833 0.0 0.0 0.0 0.0 127.0 0.5 0.5 -1
0.0 2.0 1728 86.3303 0.0 0.0 0.0 0.0 127.0 0.5 0.5 -1
0.0 2.49 2160 87.3194 0.0 0.0 0.0 0.0 127.0 0.5 0.5 -1
0.0 2.99 2592 88.3272 0.0 0.0 0.0 0.0 127.0 0.5 0.5 -1
0.0 3.49 3024 89.1249 0.0 0.0 0.0 0.0 127.0 0.5 0.5 -1
0.0 3.99 3456 89.5903 0.0 0.0 0.0 0.0 127.0 0.5 0.5 -1
0.0 4.49 3888 89.8958 0.0 0.0 0.0 0.0 127.0 0.5 0.5 -1
0.0 4.99 4320 89.9988 0.0 0.0 0.0 0.0 127.0 0.5 0.5 -1

Framework versions

  • Transformers 4.39.3
  • Pytorch 2.3.0+cu121
  • Datasets 2.19.0
  • Tokenizers 0.15.2