texttron / tevatron

Tevatron - A flexible toolkit for neural retrieval research and development.
http://tevatron.ai
Apache License 2.0
531 stars 100 forks source link

[DRAFT] Add support gradient cache for Reranker #148

Open sigridjineth opened 3 months ago

sigridjineth commented 3 months ago

https://github.com/luyug/GradCache?tab=readme-ov-file#functional-approach

MXueguang commented 3 months ago

Hi @sigridjineth, thanks for the PR, this is a very nice feature! Let me know when it is ready for review/test.