jingtaozhan / JPQ

CIKM'21: JPQ substantially improves the efficiency of Dense Retrieval with 30x compression ratio, 10x CPU speedup and 2x GPU speedup.
MIT License
49 stars 11 forks source link

return faiss index and model from run_train #5

Closed cmacdonald closed 2 years ago

cmacdonald commented 2 years ago

Jingtao, please consider this minor architectural change. This would allow easier reuse of the run_train() code.

jingtaozhan commented 2 years ago

Thank you!