salesforce / CodeT5

Home of CodeT5: Open Code LLMs for Code Understanding and Generation
https://arxiv.org/abs/2305.07922
BSD 3-Clause "New" or "Revised" License
2.68k stars 394 forks source link

Script for finetuning refine task ? #33

Closed Rishov-NightKING closed 2 years ago

Rishov-NightKING commented 2 years ago

Hi, I would try to finetune different dataset on refine task. But I cant seem to find the script for refinement task.

It will help me a lot if you enlighten me up.

Thanks in advance.

yuewang-cuhk commented 2 years ago

Hi, the fine-tuning script for all generation task is run_gen.py. Please refer to here for how to fine-tune on different datasets. Thanks.