How to finetune NLI model
mdabedr opened this issue · comments
Md Abed Rahman commented
Hi, could you please give me an overview of how to fine-tune an NLI model? Namely:
- Which file to fine-tune
- Possible prompts and commands, values of w or any relevant details
Sean commented
Hi, it is recommended to read this doc: https://github.com/SeanLee97/AnglE/tree/main/examples/NLI
I will add more docs later.
Md Abed Rahman commented
So I have to train the angle model before I can fine-tune it for the dataset I have in mind?
Sean commented
If you have prepared the NLI datasets, you can fine-tune the model using angle-trainer