├── .gitignore ├── LICENSE ├── README.md ├── docs ├── README.md └── img │ ├── approach.png │ └── icon.png ├── requirements.txt ├── scripts ├── run_cross_encoder.sh ├── run_multi_vector_biencoder.sh ├── run_single_vector_biencoder.sh └── run_single_vector_distillation.sh └── src ├── data ├── mmarco.py ├── mrtydi.py └── other.py ├── retrievers ├── cross_encoder.py ├── multi_vector_biencoder.py ├── single_vector_biencoder.py └── single_vector_distiller.py └── utils ├── ColBERT.py ├── SentenceTransformer.py ├── common.py └── loggers.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/README.md -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/img/approach.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/docs/img/approach.png -------------------------------------------------------------------------------- /docs/img/icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/docs/img/icon.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/run_cross_encoder.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/scripts/run_cross_encoder.sh -------------------------------------------------------------------------------- /scripts/run_multi_vector_biencoder.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/scripts/run_multi_vector_biencoder.sh -------------------------------------------------------------------------------- /scripts/run_single_vector_biencoder.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/scripts/run_single_vector_biencoder.sh -------------------------------------------------------------------------------- /scripts/run_single_vector_distillation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/scripts/run_single_vector_distillation.sh -------------------------------------------------------------------------------- /src/data/mmarco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/src/data/mmarco.py -------------------------------------------------------------------------------- /src/data/mrtydi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/src/data/mrtydi.py -------------------------------------------------------------------------------- /src/data/other.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/src/data/other.py -------------------------------------------------------------------------------- /src/retrievers/cross_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/src/retrievers/cross_encoder.py -------------------------------------------------------------------------------- /src/retrievers/multi_vector_biencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/src/retrievers/multi_vector_biencoder.py -------------------------------------------------------------------------------- /src/retrievers/single_vector_biencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/src/retrievers/single_vector_biencoder.py -------------------------------------------------------------------------------- /src/retrievers/single_vector_distiller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/src/retrievers/single_vector_distiller.py -------------------------------------------------------------------------------- /src/utils/ColBERT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/src/utils/ColBERT.py -------------------------------------------------------------------------------- /src/utils/SentenceTransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/src/utils/SentenceTransformer.py -------------------------------------------------------------------------------- /src/utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/src/utils/common.py -------------------------------------------------------------------------------- /src/utils/loggers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-louis/xm-retrievers/HEAD/src/utils/loggers.py --------------------------------------------------------------------------------