Skip to content

GitLab

Unverified Commit 44c340f4 authored by Yoshitomo Matsubara's avatar Yoshitomo Matsubara Committed by GitHub
Browse files

fix a bug in eval_batch_retrieval (#9089)

parent c19d0462
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/research_projects/rag/eval_rag.py
View file @ 44c340f4
...@@ -96,7 +96,7 @@ def evaluate_batch_retrieval(args, rag_model, questions): ...@@ -96,7 +96,7 @@ def evaluate_batch_retrieval(args, rag_model, questions):
)["input_ids"].to(args.device) )["input_ids"].to(args.device)
question_enc_outputs = rag_model.rag.question_encoder(retriever_input_ids) 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( result = rag_model.retriever(
retriever_input_ids, retriever_input_ids,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment