CoolFace
Modelpublic

smangrul/starcoder-3b-hugcoder

sourceHugging Facebigcode-openrail-mupdated 3y agoView on Hugging Face
0likes18downloads
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. -->

starcoder-3b-hugcoder

This model is a fine-tuned version of bigcode/starcoder2-3b on smangrul/hug_stack dataset. It achieves the following results on the evaluation set:

  • Loss: 0.5545

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.0002
  • trainbatchsize: 16
  • evalbatchsize: 16
  • seed: 11
  • gradientaccumulationsteps: 4
  • totaltrainbatch_size: 64
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lrschedulertype: cosine
  • lrschedulerwarmup_ratio: 0.1
  • training_steps: 2000

Training results

Training LossEpochStepValidation Loss
0.70190.051000.7098
0.68170.12000.6754
0.57720.153000.6445
0.58360.24000.6228
0.60570.255000.6072
0.53090.36000.5943
0.50990.357000.5862
0.51150.48000.5781
0.51030.459000.5728
0.41010.510000.5685
0.46940.5511000.5636
0.43640.612000.5605
0.45160.6513000.5577
0.42830.714000.5570
0.43240.7515000.5550
0.46930.816000.5554
0.4440.8517000.5548
0.46080.918000.5538
0.38910.9519000.5548
0.40281.020000.5545

Framework versions

  • PEFT 0.9.1.dev0
  • Transformers 4.39.0.dev0
  • Pytorch 2.1.1+cu121
  • Datasets 2.18.0
  • Tokenizers 0.15.2