CoolFace
Modelpublic

AdnanRiaz107/CodeBert-finetuned-the-stack-bash

sourceHugging Faceupdated 3y agoView on Hugging Face
0likes22downloads
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. -->

CodeBert-finetuned-the-stack-bash

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

  • Loss: 5.6895

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: 10000

Training results

Training LossEpochStepValidation Loss
8.14730.055007.4270
6.93720.110006.9409
6.1670.1515006.6749
7.97450.220006.4053
7.25760.2525006.2217
6.4510.330006.0992
6.32180.3535006.0231
6.31150.440006.0302
6.63430.4545005.8745
6.15150.550005.8281
6.39920.5555005.7614
6.84210.660005.8745
6.05420.6565005.7452
5.32060.770005.7668
6.1210.7575005.6950
6.59560.880005.6926
5.86670.8585005.6904
6.02870.990005.6803
5.84170.9595005.6747
6.97191.0100005.6895

Framework versions

  • Transformers 4.36.0.dev0
  • Pytorch 2.1.0+cu118
  • Datasets 2.15.0
  • Tokenizers 0.15.0