path	# lines of code
dpr_scale/generate_embeddings.py	21
dpr_scale/datamodule/__init__.py	1
dpr_scale/datamodule/dpr.py	207
dpr_scale/generate_query_embeddings.py	23
dpr_scale/run_retrieval_multiset.py	134
dpr_scale/optim/__init__.py	1
dpr_scale/optim/madgrad.py	121
dpr_scale/__init__.py	1
dpr_scale/utils/__init__.py	1
dpr_scale/utils/ccnews_stats.py	87
dpr_scale/utils/tune_spar_weights.py	160
dpr_scale/utils/prep_wiki.py	117
dpr_scale/utils/utils.py	89
dpr_scale/utils/prep_wiki_exp.py	179
dpr_scale/utils/prep_ccnews.py	169
dpr_scale/models/hf_model.py	30
dpr_scale/models/hf_bert.py	19
dpr_scale/models/__init__.py	1
dpr_scale/data_prep/prep_conv_datasets.py	105
dpr_scale/run_retrieval_fb.py	127
dpr_scale/run_retrieval.py	90
dpr_scale/transforms/hf_transform.py	27
dpr_scale/transforms/dpr_transform.py	220
dpr_scale/transforms/hf_bert.py	28
dpr_scale/transforms/__init__.py	1
dpr_scale/task/dpr_eval_task.py	67
dpr_scale/task/__init__.py	1
dpr_scale/task/dpr_task.py	260
dpr_scale/main.py	33
dpr_scale/conf/msmarco.yaml	31
dpr_scale/conf/orcas.yaml	35
dpr_scale/conf/datamodule/generate.yaml	5
dpr_scale/conf/datamodule/default.yaml	9
dpr_scale/conf/config.py	24
dpr_scale/conf/nq_eval.yaml	5
dpr_scale/conf/ubuntuv2.yaml	37
dpr_scale/conf/__init__.py	1
dpr_scale/conf/nq.yaml	29
dpr_scale/conf/task/dpr.yaml	6
dpr_scale/conf/task/optim/madgrad.yaml	5
dpr_scale/conf/task/optim/lamb.yaml	7
dpr_scale/conf/task/optim/adamw.yaml	8
dpr_scale/conf/task/model/hf_model.yaml	4
dpr_scale/conf/task/model/bert_encoder_default.yaml	1
dpr_scale/conf/task/transform/hf_transform.yaml	3
dpr_scale/conf/task/transform/bert_transform_default.yaml	1
dpr_scale/conf/checkpoint_callback/default.yaml	7
dpr_scale/conf/convai2.yaml	36
dpr_scale/conf/dstc7.yaml	39
dpr_scale/conf/nq_roberta.yaml	29
dpr_scale/conf/reddit.yaml	36
dpr_scale/conf/trainer/gpu_1_host.yaml	7
dpr_scale/conf/trainer/slurm.yaml	25
dpr_scale/conf/wiki_ict.yaml	32
dpr_scale/conf/ccnews_ict.yaml	30
dpr_scale/eval_dpr.py	180
dpr_scale/msmarco_eval.py	155