Doohae / DenseRetrieval

Implementation of DPR, GradCache, DSI etc.

Dense Retrieval

Run Training in Google Colab

https://colab.research.google.com/drive/1b2VYvxgLNbsHKvutimAN6n--Vh2WzI40?usp=sharing

TODO

1. MS-Marco Data Processing
2. Dense Passage Retrieval
3. GradCache (+ Condenser)
4. Differentiable Search Index
5. HuggingFace Accelerate (TPU, multi-GPU)