CoolFace
Modelpublic

lmeninato/flan-t5-small-codesearchnet-python

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

flan-t5-small-codesearchnet-python

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

  • Loss: 0.0764
  • Bleu: 0.0349
  • Rouge1: 0.6244
  • Rouge2: 0.6055
  • Avg Length: 16.9912

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.06360.03640.62530.607617.029
5.51662.07500.05530.03510.62590.608116.9996
0.04853.011250.05370.03510.62580.608316.99
0.04094.015000.05240.03510.62580.608216.9942
0.04095.018750.05240.03510.62610.608616.997
0.03456.022500.05260.03510.62580.608116.9936
0.03037.026250.05330.0350.62540.607616.991
0.02568.030000.05660.0350.62570.607416.9964
0.02569.033750.05920.03490.62530.607416.998
0.020510.037500.06120.03510.62550.607316.9932
0.018511.041250.06390.0350.62570.607916.996
0.015712.045000.06980.0350.6250.606416.9944
0.015713.048750.07200.0350.62460.606216.991
0.013114.052500.07450.0350.62470.606216.9986
0.012815.056250.07640.03490.62440.605516.9912

Framework versions

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