fix: #5

boostcampaitech6 · Feb 18, 2024 · a77710c · a77710c
1 parent 0669e29
commit a77710c
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/code/inference_dpr.py b/code/inference_dpr.py
@@ -129,7 +129,6 @@ def run_dense_retrieval(
     )
     retriever = DenseRetrieval(
         args=args,
-        dataset=datasets,
         num_neg=3,
         tokenizer=tokenizer,
         p_encoder=p_encoder,