CoolFace
Modelpublic

lmeninato/t5-small-codesearchnet-multilang-python-java-javascript-go

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
0likes19downloads
README.md80 linesDownload Raw Back to root
1---2license: apache-2.03tags:4- generated_from_trainer5metrics:6- bleu7- rouge8model-index:9- name: t5-small-codesearchnet-multilang-python-java-javascript-go10  results: []11---12 13<!-- This model card has been generated automatically according to the information the Trainer had access to. You14should probably proofread and complete it, then remove this comment. -->15 16# t5-small-codesearchnet-multilang-python-java-javascript-go17 18This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on the None dataset.19It achieves the following results on the evaluation set:20- Loss: 0.595521- Bleu: 0.00922- Rouge1: 0.232123- Rouge2: 0.083124- Avg Length: 16.619225 26## Model description27 28More information needed29 30## Intended uses & limitations31 32More information needed33 34## Training and evaluation data35 36More information needed37 38## Training procedure39 40### Training hyperparameters41 42The following hyperparameters were used during training:43- learning_rate: 5e-0544- train_batch_size: 845- eval_batch_size: 846- seed: 4247- gradient_accumulation_steps: 1048- total_train_batch_size: 8049- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-0850- lr_scheduler_type: linear51- num_epochs: 1552 53### Training results54 55| Training Loss | Epoch | Step | Validation Loss | Bleu   | Rouge1 | Rouge2 | Avg Length |56|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:----------:|57| No log        | 1.0   | 375  | 0.7349          | 0.0028 | 0.1562 | 0.0364 | 16.436     |58| 2.3117        | 2.0   | 750  | 0.6613          | 0.0066 | 0.1818 | 0.0531 | 16.824     |59| 0.6755        | 3.0   | 1125 | 0.6233          | 0.007  | 0.1957 | 0.0594 | 16.931     |60| 0.5998        | 4.0   | 1500 | 0.6023          | 0.0082 | 0.202  | 0.063  | 16.7154    |61| 0.5998        | 5.0   | 1875 | 0.5925          | 0.0096 | 0.2154 | 0.0703 | 16.5468    |62| 0.5511        | 6.0   | 2250 | 0.5728          | 0.0091 | 0.2213 | 0.0774 | 15.7216    |63| 0.5147        | 7.0   | 2625 | 0.5670          | 0.0111 | 0.2311 | 0.0815 | 16.6658    |64| 0.4861        | 8.0   | 3000 | 0.5628          | 0.0089 | 0.2217 | 0.077  | 17.038     |65| 0.4861        | 9.0   | 3375 | 0.5598          | 0.0103 | 0.2311 | 0.0825 | 16.362     |66| 0.4526        | 10.0  | 3750 | 0.5589          | 0.0083 | 0.232  | 0.086  | 15.4298    |67| 0.4329        | 11.0  | 4125 | 0.5649          | 0.0098 | 0.2349 | 0.0839 | 16.5468    |68| 0.4102        | 12.0  | 4500 | 0.5633          | 0.0098 | 0.2366 | 0.0867 | 16.4136    |69| 0.4102        | 13.0  | 4875 | 0.5841          | 0.01   | 0.2385 | 0.0869 | 15.9864    |70| 0.3841        | 14.0  | 5250 | 0.5777          | 0.0128 | 0.2437 | 0.0894 | 16.842     |71| 0.3673        | 15.0  | 5625 | 0.5955          | 0.009  | 0.2321 | 0.0831 | 16.6192    |72 73 74### Framework versions75 76- Transformers 4.28.177- Pytorch 2.0.0+cu11878- Datasets 2.12.079- Tokenizers 0.13.380