CoolFace
Modelpublic

airKlizz/mt5-base-wikinewssum-all-languages

sourceHugging Faceapache-2.0updated 5y agoView on Hugging Face
1likes32downloads
Model Card

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

mt5-base-wikinewssum-all-languages

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

  • —Loss: 2.2454
  • —Rouge1: 8.3826
  • —Rouge2: 3.5524
  • —Rougel: 6.8656
  • —Rougelsum: 7.8362

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: 5.6e-05
  • —trainbatchsize: 4
  • —evalbatchsize: 4
  • —seed: 42
  • —gradientaccumulationsteps: 2
  • —totaltrainbatch_size: 8
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —num_epochs: 8

Training results

Training LossEpochStepValidation LossRouge1Rouge2RougelRougelsum
No log1.034672.40348.03633.24846.54097.477
No log2.069342.32768.10543.29056.57657.5687
No log3.0104012.29768.1693.42726.65977.6435
No log4.0138682.27958.29413.53536.78817.7664
2.80575.0173352.26218.33023.55996.82387.7928
2.80576.0208022.25478.38183.58866.86727.844
2.80577.0242692.24728.38093.56966.85757.8327
2.80578.0277362.24548.38263.55246.86567.8362

Framework versions

  • —Transformers 4.13.0
  • —Pytorch 1.10.1
  • —Datasets 1.16.1
  • —Tokenizers 0.10.3