├── .gitignore ├── LICENSE ├── README.md ├── examples ├── dense │ ├── chinese-dureader │ │ ├── adapt_domain.md │ │ ├── inference.md │ │ ├── inference_baseline.md │ │ ├── train_baseline.md │ │ └── train_rem.md │ └── english-marco │ │ ├── inference.md │ │ ├── inference_baseline.md │ │ ├── train_baseline.md │ │ └── train_rem.md ├── domain_adapt │ ├── chinese-dureader │ │ ├── adapt_to_new_domain.md │ │ └── train_source_dam.md │ └── english-marco │ │ ├── adapt_to_new_domain.md │ │ └── train_source_dam.md └── prepare_dataset │ ├── chinese-dureader │ ├── README.md │ ├── download_dureader.sh │ ├── process_cmedqav2.py │ └── process_dureader.py │ └── english-marco │ ├── README.md │ ├── prepare_lotte.py │ ├── prepare_marco_hardneg.py │ ├── prepare_msmarco.sh │ └── prepare_trec-covid.py ├── figures ├── NDCG@10.png ├── R@1000.png ├── colbert-ndcg@10.png ├── colbert-recall@1000.png ├── ddr-modeling.png └── dr-modeling.png ├── setup.py └── src └── disentangled_retriever ├── __init__.py ├── adapt ├── __init__.py └── run_adapt_with_mlm.py ├── adapter_arg.py ├── colbert ├── __init__.py ├── evaluate │ ├── index.py │ ├── index_faiss.py │ ├── model_inference.py │ └── retrieve.py ├── finetune │ ├── __init__.py │ ├── contrast_utils.py │ ├── distill_utils.py │ ├── run_contrast.py │ └── run_distill.py └── modeling │ ├── __init__.py │ └── bert_colbert.py ├── dense ├── __init__.py ├── evaluate │ ├── __init__.py │ ├── index_utils.py │ └── run_eval.py ├── finetune │ ├── __init__.py │ ├── contrast_utils.py │ ├── distill_utils.py │ ├── run_contrast.py │ ├── run_distill.py │ └── validate_utils.py └── modeling │ ├── __init__.py │ ├── bert_dense.py │ ├── distilbert_dense.py │ ├── roberta_dense.py │ └── utils.py ├── evaluate.py ├── rerank ├── __init__.py ├── evaluate │ ├── __init__.py │ ├── eval_utils.py │ └── run_eval.py └── finetune │ ├── __init__.py │ ├── run_train.py │ └── train_utils.py ├── splade ├── __init__.py ├── evaluate │ ├── index_utils.py │ └── run_eval.py ├── finetune │ ├── contrast_utils.py │ ├── distill_utils.py │ ├── run_contrast.py │ └── run_distill.py └── modeling │ ├── __init__.py │ └── bert_splade.py └── unicoil ├── __init__.py ├── evaluate ├── index_utils.py └── run_eval.py ├── finetune ├── contrast_utils.py ├── distill_utils.py ├── run_contrast.py └── run_distill.py └── modeling ├── __init__.py └── bert_unicoil.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/README.md -------------------------------------------------------------------------------- /examples/dense/chinese-dureader/adapt_domain.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/dense/chinese-dureader/adapt_domain.md -------------------------------------------------------------------------------- /examples/dense/chinese-dureader/inference.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/dense/chinese-dureader/inference.md -------------------------------------------------------------------------------- /examples/dense/chinese-dureader/inference_baseline.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/dense/chinese-dureader/inference_baseline.md -------------------------------------------------------------------------------- /examples/dense/chinese-dureader/train_baseline.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/dense/chinese-dureader/train_baseline.md -------------------------------------------------------------------------------- /examples/dense/chinese-dureader/train_rem.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/dense/chinese-dureader/train_rem.md -------------------------------------------------------------------------------- /examples/dense/english-marco/inference.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/dense/english-marco/inference.md -------------------------------------------------------------------------------- /examples/dense/english-marco/inference_baseline.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/dense/english-marco/inference_baseline.md -------------------------------------------------------------------------------- /examples/dense/english-marco/train_baseline.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/dense/english-marco/train_baseline.md -------------------------------------------------------------------------------- /examples/dense/english-marco/train_rem.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/dense/english-marco/train_rem.md -------------------------------------------------------------------------------- /examples/domain_adapt/chinese-dureader/adapt_to_new_domain.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/domain_adapt/chinese-dureader/adapt_to_new_domain.md -------------------------------------------------------------------------------- /examples/domain_adapt/chinese-dureader/train_source_dam.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/domain_adapt/chinese-dureader/train_source_dam.md -------------------------------------------------------------------------------- /examples/domain_adapt/english-marco/adapt_to_new_domain.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/domain_adapt/english-marco/adapt_to_new_domain.md -------------------------------------------------------------------------------- /examples/domain_adapt/english-marco/train_source_dam.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/domain_adapt/english-marco/train_source_dam.md -------------------------------------------------------------------------------- /examples/prepare_dataset/chinese-dureader/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/prepare_dataset/chinese-dureader/README.md -------------------------------------------------------------------------------- /examples/prepare_dataset/chinese-dureader/download_dureader.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/prepare_dataset/chinese-dureader/download_dureader.sh -------------------------------------------------------------------------------- /examples/prepare_dataset/chinese-dureader/process_cmedqav2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/prepare_dataset/chinese-dureader/process_cmedqav2.py -------------------------------------------------------------------------------- /examples/prepare_dataset/chinese-dureader/process_dureader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/prepare_dataset/chinese-dureader/process_dureader.py -------------------------------------------------------------------------------- /examples/prepare_dataset/english-marco/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/prepare_dataset/english-marco/README.md -------------------------------------------------------------------------------- /examples/prepare_dataset/english-marco/prepare_lotte.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/prepare_dataset/english-marco/prepare_lotte.py -------------------------------------------------------------------------------- /examples/prepare_dataset/english-marco/prepare_marco_hardneg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/prepare_dataset/english-marco/prepare_marco_hardneg.py -------------------------------------------------------------------------------- /examples/prepare_dataset/english-marco/prepare_msmarco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/prepare_dataset/english-marco/prepare_msmarco.sh -------------------------------------------------------------------------------- /examples/prepare_dataset/english-marco/prepare_trec-covid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/examples/prepare_dataset/english-marco/prepare_trec-covid.py -------------------------------------------------------------------------------- /figures/NDCG@10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/figures/NDCG@10.png -------------------------------------------------------------------------------- /figures/R@1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/figures/R@1000.png -------------------------------------------------------------------------------- /figures/colbert-ndcg@10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/figures/colbert-ndcg@10.png -------------------------------------------------------------------------------- /figures/colbert-recall@1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/figures/colbert-recall@1000.png -------------------------------------------------------------------------------- /figures/ddr-modeling.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/figures/ddr-modeling.png -------------------------------------------------------------------------------- /figures/dr-modeling.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/figures/dr-modeling.png -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/setup.py -------------------------------------------------------------------------------- /src/disentangled_retriever/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/disentangled_retriever/adapt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/disentangled_retriever/adapt/run_adapt_with_mlm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/adapt/run_adapt_with_mlm.py -------------------------------------------------------------------------------- /src/disentangled_retriever/adapter_arg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/adapter_arg.py -------------------------------------------------------------------------------- /src/disentangled_retriever/colbert/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/disentangled_retriever/colbert/evaluate/index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/colbert/evaluate/index.py -------------------------------------------------------------------------------- /src/disentangled_retriever/colbert/evaluate/index_faiss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/colbert/evaluate/index_faiss.py -------------------------------------------------------------------------------- /src/disentangled_retriever/colbert/evaluate/model_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/colbert/evaluate/model_inference.py -------------------------------------------------------------------------------- /src/disentangled_retriever/colbert/evaluate/retrieve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/colbert/evaluate/retrieve.py -------------------------------------------------------------------------------- /src/disentangled_retriever/colbert/finetune/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/disentangled_retriever/colbert/finetune/contrast_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/colbert/finetune/contrast_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/colbert/finetune/distill_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/colbert/finetune/distill_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/colbert/finetune/run_contrast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/colbert/finetune/run_contrast.py -------------------------------------------------------------------------------- /src/disentangled_retriever/colbert/finetune/run_distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/colbert/finetune/run_distill.py -------------------------------------------------------------------------------- /src/disentangled_retriever/colbert/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/colbert/modeling/__init__.py -------------------------------------------------------------------------------- /src/disentangled_retriever/colbert/modeling/bert_colbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/colbert/modeling/bert_colbert.py -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/evaluate/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/evaluate/index_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/dense/evaluate/index_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/evaluate/run_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/dense/evaluate/run_eval.py -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/finetune/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/finetune/contrast_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/dense/finetune/contrast_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/finetune/distill_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/dense/finetune/distill_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/finetune/run_contrast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/dense/finetune/run_contrast.py -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/finetune/run_distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/dense/finetune/run_distill.py -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/finetune/validate_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/dense/finetune/validate_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/dense/modeling/__init__.py -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/modeling/bert_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/dense/modeling/bert_dense.py -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/modeling/distilbert_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/dense/modeling/distilbert_dense.py -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/modeling/roberta_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/dense/modeling/roberta_dense.py -------------------------------------------------------------------------------- /src/disentangled_retriever/dense/modeling/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/dense/modeling/utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/evaluate.py -------------------------------------------------------------------------------- /src/disentangled_retriever/rerank/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/disentangled_retriever/rerank/evaluate/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/disentangled_retriever/rerank/evaluate/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/rerank/evaluate/eval_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/rerank/evaluate/run_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/rerank/evaluate/run_eval.py -------------------------------------------------------------------------------- /src/disentangled_retriever/rerank/finetune/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/rerank/finetune/__init__.py -------------------------------------------------------------------------------- /src/disentangled_retriever/rerank/finetune/run_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/rerank/finetune/run_train.py -------------------------------------------------------------------------------- /src/disentangled_retriever/rerank/finetune/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/rerank/finetune/train_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/splade/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/disentangled_retriever/splade/evaluate/index_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/splade/evaluate/index_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/splade/evaluate/run_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/splade/evaluate/run_eval.py -------------------------------------------------------------------------------- /src/disentangled_retriever/splade/finetune/contrast_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/splade/finetune/contrast_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/splade/finetune/distill_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/splade/finetune/distill_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/splade/finetune/run_contrast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/splade/finetune/run_contrast.py -------------------------------------------------------------------------------- /src/disentangled_retriever/splade/finetune/run_distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/splade/finetune/run_distill.py -------------------------------------------------------------------------------- /src/disentangled_retriever/splade/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/splade/modeling/__init__.py -------------------------------------------------------------------------------- /src/disentangled_retriever/splade/modeling/bert_splade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/splade/modeling/bert_splade.py -------------------------------------------------------------------------------- /src/disentangled_retriever/unicoil/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/disentangled_retriever/unicoil/evaluate/index_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/unicoil/evaluate/index_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/unicoil/evaluate/run_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/unicoil/evaluate/run_eval.py -------------------------------------------------------------------------------- /src/disentangled_retriever/unicoil/finetune/contrast_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/unicoil/finetune/contrast_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/unicoil/finetune/distill_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/unicoil/finetune/distill_utils.py -------------------------------------------------------------------------------- /src/disentangled_retriever/unicoil/finetune/run_contrast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/unicoil/finetune/run_contrast.py -------------------------------------------------------------------------------- /src/disentangled_retriever/unicoil/finetune/run_distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/unicoil/finetune/run_distill.py -------------------------------------------------------------------------------- /src/disentangled_retriever/unicoil/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/unicoil/modeling/__init__.py -------------------------------------------------------------------------------- /src/disentangled_retriever/unicoil/modeling/bert_unicoil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jingtaozhan/disentangled-retriever/HEAD/src/disentangled_retriever/unicoil/modeling/bert_unicoil.py --------------------------------------------------------------------------------