From 44c340f45f1b2ef0c477a13ed94b0ca0f2bdfab9 Mon Sep 17 00:00:00 2001 From: Yoshitomo Matsubara Date: Tue, 15 Dec 2020 05:46:55 -0800 Subject: [PATCH] fix a bug in eval_batch_retrieval (#9089) --- examples/research_projects/rag/eval_rag.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/research_projects/rag/eval_rag.py b/examples/research_projects/rag/eval_rag.py index d479537ff3..dfedab602b 100644 --- a/examples/research_projects/rag/eval_rag.py +++ b/examples/research_projects/rag/eval_rag.py @@ -96,7 +96,7 @@ def evaluate_batch_retrieval(args, rag_model, questions): )["input_ids"].to(args.device) question_enc_outputs = rag_model.rag.question_encoder(retriever_input_ids) - question_enc_pool_output = question_enc_outputs.pooler_output + question_enc_pool_output = question_enc_outputs[0] result = rag_model.retriever( retriever_input_ids,