search

NLPJCL / RAG-Retrieval

Unify Efficient Fine-tuning of RAG Retrieval, including Embedding, ColBERT, ReRanker.
MIT License
506 stars 45 forks source link

Adjust a little code for MRL #20

Closed BUAADreamer closed 5 months ago

BUAADreamer commented 5 months ago
  1. modify parser to receive mrl_dims as a param. usage: --mrl_dims 1,2,4
  2. implement mrl loss for kl pair loss branch.