CoolFace
Modelpublic

14rohnisingh/peft-starcoder-lora-a100

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

peft-starcoder-lora-a100

This model is a fine-tuned version of bigcode/starcoderbase-1b on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 0.4433

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

Training results

Training LossEpochStepValidation Loss
0.450.051000.4334
0.4130.12000.4235
0.3960.153000.4189
0.37020.24000.4082
0.3560.255000.4072
0.34290.36000.4129
0.33750.357000.4136
0.31810.48000.4080
0.30570.459000.4187
0.29560.510000.4214
0.27610.5511000.4261
0.28050.612000.4275
0.26970.6513000.4294
0.25560.714000.4338
0.25780.7515000.4339
0.24270.816000.4401
0.2450.8517000.4423
0.2480.918000.4435
0.23770.9519000.4436
0.24661.020000.4433

Framework versions

  • PEFT 0.11.1
  • Transformers 4.40.2
  • Pytorch 2.2.1+cu121
  • Datasets 2.19.1
  • Tokenizers 0.19.1