CoolFace
Modelpublic

Smaraa/pegasus-text-simplification_1e4_adafactor_wikilarge_20epici

sourceHugging Faceupdated 3y agoView on Hugging Face
0likes5downloads
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. -->

pegasus-text-simplification1e4adafactorwikilarge20epici

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

  • —Loss: 0.4783
  • —Rouge1: 72.0576
  • —Rouge2: 53.6457
  • —Rougel: 66.9028
  • —Rougelsum: 66.9302
  • —Gen Len: 21.9895

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
  • —trainbatchsize: 128
  • —evalbatchsize: 128
  • —seed: 42
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —num_epochs: 20

Training results

Training LossEpochStepValidation LossRouge1Rouge2RougelRougelsumGen Len
0.51871.011630.450273.668754.949968.254468.256823.6492
0.47982.023260.447273.846555.541568.529268.545723.4241
0.45423.034890.445674.003155.52168.702368.687123.0838
0.43764.046520.447673.423955.059168.193168.183322.9529
0.42265.058150.448473.185954.755867.884367.923422.6702
0.41086.069780.451772.860254.381367.576167.626622.356
0.39937.081410.453273.041354.725467.780267.806522.5131
0.3918.093040.459072.944854.376367.68967.671222.5079
0.38449.0104670.457972.620454.10767.390367.386322.2775
0.377310.0116300.460672.265153.643266.995867.019922.2251
0.367111.0127930.463372.730854.254467.394567.472622.0
0.360512.0139560.467572.554853.83467.244767.285522.555
0.358813.0151190.468671.968153.299566.729866.77922.089
0.35314.0162820.469471.954453.413766.758566.791522.1099
0.347415.0174450.473872.000653.335166.715366.715722.1728
0.344116.0186080.474972.018853.524266.755666.797722.0733
0.343117.0197710.475671.9153.095566.561266.503822.1361
0.338518.0209340.476971.970253.583266.91566.943622.0157
0.338419.0220970.477872.056353.715366.999867.06122.0314
0.336820.0232600.478372.057653.645766.902866.930221.9895

Framework versions

  • —Transformers 4.30.2
  • —Pytorch 2.0.1+cu118
  • —Datasets 2.13.1
  • —Tokenizers 0.13.3