CoolFace
Modelpublic

convaise-idp/flan-t5-base-finetuned-length_control_token

sourceHugging Faceapache-2.0updated 4y agoView on Hugging Face
0likes18downloads
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. -->

flan-t5-base-finetuned-lengthcontroltoken

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

  • —Loss: 1.0276
  • —Sacrebleu: 16.2445

Model description

This model was trained on a dataset called PWKP-GPT3-LENGTH-CONTROL-40BUCKETS. The dataset contains 30k instances taken from PWKP, then processed through GPT3 to obtain simplifications. The 30k instances come from: 10k which were supposed to generate very long simplifications, 10k which were supposed to generate very short simplifications, and 10k without specifying the simplicity level. The model does not sucessfuly work on these buckets. There exists another dataset, the PWKP-GPT3-LENGTH-CONTROL-4BUCKETS, but it was never trained on something. Those buckets are also rather unbalanced.

The idea comes from Controllable Sentence Simplification Louis Martin, https://arxiv.org/pdf/1910.02677.pdf

It was fine-tuned on the FLAN-T5-base model.

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

Training results

Training LossEpochStepValidation LossSacrebleu
1.32571.017821.090615.4208
1.17182.035641.064815.5358
1.09723.053461.048415.8113
1.04724.071281.039416.0159
1.00925.089101.030516.1341
0.98586.0106921.027616.2445

Framework versions

  • —Transformers 4.26.1
  • —Pytorch 1.13.1+cu117
  • —Datasets 2.10.1
  • —Tokenizers 0.13.2