CoolFace
Modelpublic

devika-tiwari/gpt2_small_expandedbabyLM_25M_43

sourceHugging Faceupdated 1mo agoView on Hugging Face
0likes39downloads
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. -->

gpt2smallexpandedbabyLM25M43

This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 2.9758

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.0001
  • trainbatchsize: 256
  • evalbatchsize: 256
  • seed: 43
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lrschedulertype: linear
  • lrschedulerwarmup_steps: 4000
  • num_epochs: 20

Training results

Training LossEpochStepValidation Loss
4.65131.012724.0029
4.03092.025443.5163
3.71673.038163.3070
3.54464.050883.1842
3.43045.063603.1125
3.33956.076323.0654
3.27667.089043.0325
3.21888.0101763.0151
3.15579.0114482.9991
3.110810.0127202.9876
3.077711.0139922.9816
3.02912.0152642.9759
2.992213.0165362.9759
2.959814.0178082.9758
2.928415.0190802.9768

Framework versions

  • Transformers 4.30.2
  • Pytorch 2.11.0+cu130
  • Datasets 4.1.1
  • Tokenizers 0.13.3