CoolFace
Modelpublic

lmeninato/t5-small-codesearchnet-python

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
0likes17downloads
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. -->

t5-small-codesearchnet-python

This model is a fine-tuned version of t5-small on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 0.0785
  • Bleu: 0.035
  • Rouge1: 0.6257
  • Rouge2: 0.6078
  • Avg Length: 16.9954

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: 8
  • evalbatchsize: 8
  • seed: 42
  • gradientaccumulationsteps: 10
  • totaltrainbatch_size: 80
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lrschedulertype: linear
  • num_epochs: 15

Training results

Training LossEpochStepValidation LossBleuRouge1Rouge2Avg Length
No log1.03750.08010.03580.61740.617.1074
1.60662.07500.06740.0360.62490.606817.0262
0.05843.011250.06320.03510.62550.607516.9962
0.04844.015000.06050.03510.62510.607117.003
0.04845.018750.05960.0350.62550.607517.0012
0.04186.022500.06020.0350.62580.60816.9958
0.03777.026250.05930.03510.62590.607917.0004
0.0338.030000.06180.0350.62570.607817.0032
0.0339.033750.06370.0350.62570.607816.998
0.02810.037500.06450.0350.62570.607916.9984
0.025511.041250.06500.0350.62550.607817.0008
0.022612.045000.07480.0350.62540.607616.9976
0.022613.048750.07140.0350.62560.607916.9954
0.01914.052500.07470.03490.62530.607716.994
0.017215.056250.07850.0350.62570.607816.9954

Framework versions

  • Transformers 4.28.1
  • Pytorch 2.0.0+cu118
  • Datasets 2.12.0
  • Tokenizers 0.13.3