SeanLee97 / AnglE

Train and Infer Powerful Sentence Embeddings with AnglE | 🔥 SOTA on STS and MTEB Leaderboard

Home Page:https://arxiv.org/abs/2309.12871

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to finetune NLI model

mdabedr opened this issue · comments

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
commented

Hi, it is recommended to read this doc: https://github.com/SeanLee97/AnglE/tree/main/examples/NLI

I will add more docs later.

So I have to train the angle model before I can fine-tune it for the dataset I have in mind?

commented

If you have prepared the NLI datasets, you can fine-tune the model using angle-trainer