CoolFace
Modelpublic

pminhyung12/gpt2-base-v0

sourceHugging Facemitupdated 3y agoView on Hugging Face
0likes12downloads
Model Card

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

pminhyung12/gpt2-base-v0

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

  • Train Loss: 5.1058
  • Validation Loss: 6.2552
  • Epoch: 7

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:

  • optimizer: {'name': 'AdamW', 'weightdecay': 0.004, 'clipnorm': None, 'globalclipnorm': None, 'clipvalue': None, 'useema': False, 'emamomentum': 0.99, 'emaoverwritefrequency': None, 'jitcompile': False, 'islegacyoptimizer': False, 'learningrate': 1e-04, 'beta1': 0.9, 'beta2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
  • trainingprecision: mixedbfloat16

Training results

Train LossValidation LossEpoch
0.91280.95150
0.61720.89651
0.57130.87632
1.20881.83003
3.68875.87004
2.89203.64225
3.80458.95556
5.10586.25527

Framework versions

  • Transformers 4.26.1
  • TensorFlow 2.12.0
  • Datasets 2.9.0
  • Tokenizers 0.13.2