CoolFace
Modelpublic

MarkGG/Romance-cleaned-1

sourceHugging Facemitupdated 4y agoView on Hugging Face
0likes19downloads
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. -->

Romance-cleaned-1

This model is a fine-tuned version of gpt2 on the None dataset. It achieves the following results on the evaluation set:

  • —Loss: 4.7175

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: 32
  • —evalbatchsize: 32
  • —seed: 42
  • —gradientaccumulationsteps: 8
  • —totaltrainbatch_size: 256
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: cosine
  • —lrschedulerwarmup_steps: 1000
  • —num_epochs: 50
  • —mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation Loss
No log0.97299.9497
No log1.97589.1816
No log2.97878.5947
No log3.971168.2217
No log4.971457.8354
No log5.971747.5075
No log6.972037.2112
No log7.972326.9077
No log8.972616.5994
No log9.972906.3077
No log10.973196.0416
No log11.973485.8126
No log12.973775.6197
No log13.974065.4789
No log14.974355.3665
No log15.974645.2738
No log16.974935.1942
No log17.975225.1382
No log18.975515.0784
No log19.975805.0347
No log20.976094.9873
No log21.976384.9514
No log22.976674.9112
No log23.976964.8838
No log24.977254.8468
No log25.977544.8221
No log26.977834.7996
No log27.978124.7815
No log28.978414.7606
No log29.978704.7394
No log30.978994.7167
No log31.979284.7140
No log32.979574.6910
No log33.979864.6844
No log34.9710154.6765
No log35.9710444.6687
No log36.9710734.6721
No log37.9711024.6724
No log38.9711314.6629
No log39.9711604.6772
No log40.9711894.6795
No log41.9712184.6788
No log42.9712474.6832
No log43.9712764.6954
No log44.9713054.7009
No log45.9713344.7082
No log46.9713634.7140
No log47.9713924.7158
No log48.9714214.7181
No log49.9714504.7175

Framework versions

  • —Transformers 4.24.0
  • —Pytorch 1.12.1+cu113
  • —Datasets 2.6.1
  • —Tokenizers 0.13.1