search

bdqnghi / infercode

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

Training the model #16

Closed shaileshj2803 closed 2 years ago

shaileshj2803 commented 3 years ago

Awesome work. I am trying to train the model on my own data. Where is the model saved after fine tuning? i tried example from test script. Also for inference how can i load the saved model? Thanks a lot

shaileshj2803 commented 2 years ago

no longer needed

Avv22 commented 2 years ago

@shaileshj2803. Hello!

I am having same trouble, I am trying to get their pretrained model, but it seems it's for Java and not for Python, should I train the model from scratch please?