CoolFace
Modelpublic

aiswaryasankar/santacoder-finetuned-the-stack-bash

sourceHugging Facebigcode-openrail-mupdated 3y agoView on Hugging Face
0likes39downloads
Model Card

santacoder-finetuned-the-stack-bash

This model is a fine-tuned version of bigcode/santacoder on the dbrief dataset. It achieves the following results on the evaluation set:

  • Loss: 0.9406

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: 1
  • evalbatchsize: 1
  • seed: 42
  • gradientaccumulationsteps: 4
  • totaltrainbatch_size: 4
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lrschedulertype: cosine
  • lrschedulerwarmup_steps: 100
  • training_steps: 5000

Training results

Training LossEpochStepValidation Loss
3.17680.15001.6939
2.84510.210002.0221
3.01410.315002.1031
2.73060.420002.1053
2.33850.525002.0770
1.9810.630000.9364
1.66870.735000.9637
1.26630.840000.9459
1.74850.945000.9428
1.90061.050000.9406

Framework versions

  • Transformers 4.31.0
  • Pytorch 2.0.1+cu118
  • Datasets 2.14.0
  • Tokenizers 0.13.3