CoolFace
Modelpublic

anushkat/DistilGPT2-Beatles-Spotify-FinalVersion

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

DistilGPT2-Beatles-Spotify-FinalVersion

This model is a fine-tuned version of wvangils/DistilGPT2-Beatles-Lyrics-finetuned on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 1.8839

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

Training results

Training LossEpochStepValidation Loss
2.11250.0251.9245
2.10350.0501.9315
2.10640.0751.9184
2.08740.01001.9101
2.09480.01251.9143
2.10880.01501.8953
2.06120.011751.9130
2.08280.012001.9046
2.06760.012251.8950
2.0720.012501.9092
2.07480.012751.8814
2.07690.013001.8948
2.0680.013251.8921
2.05130.013501.8904
2.06860.013751.8794
2.07020.014001.8800
2.0590.014251.8780
2.05620.014501.8841
2.07050.014751.8824
2.06750.025001.8839

Framework versions

  • PEFT 0.8.2
  • Transformers 4.38.1
  • Pytorch 1.13.1
  • Datasets 2.17.0
  • Tokenizers 0.15.2