CoolFace
Modelpublic

cppbear/codeparrot-ds

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

codeparrot-ds

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

  • Loss: 1.0585

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.0005
  • trainbatchsize: 32
  • evalbatchsize: 32
  • seed: 42
  • gradientaccumulationsteps: 8
  • totaltrainbatch_size: 256
  • optimizer: Use adamwtorch with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
  • lrschedulertype: cosine
  • lrschedulerwarmup_steps: 1000
  • num_epochs: 1
  • mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation Loss
2.54980.076650001.7429
1.66710.1533100001.5218
1.5190.2299150001.4193
1.44050.3065200001.3529
1.38110.3832250001.3025
1.33020.4598300001.2536
1.28480.5365350001.2101
1.23870.6131400001.1680
1.19820.6897450001.1297
1.16050.7664500001.0966
1.13370.8430550001.0736
1.11240.9196600001.0614
1.10510.9963650001.0585

Framework versions

  • Transformers 4.52.4
  • Pytorch 2.8.0.dev20250529+rocm6.4
  • Datasets 3.6.0
  • Tokenizers 0.21.1