bdqnghi / infercode

[ICSE 2021] - InferCode: Self-Supervised Learning of Code Representations by Predicting Subtrees
89 stars 24 forks source link

Infercode fine-tuning #30

Open ElchaabiMohamed opened 2 years ago

ElchaabiMohamed commented 2 years ago

Thank you for our awesome work, i can run it successfully. Now i want to fine-tune Infercode model for a downstream task (classification of programs by their functionalities). May I know whether it is the right way to go? If so, would you mind sharing some ideas/insights about how the architecture of the new model looks like?