SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix a bug in eval_batch_retrieval (#9089)

Browse Source
This commit is contained in:
Yoshitomo Matsubara
2020-12-15 05:46:55 -08:00
committed by GitHub
parent c19d04623e
commit 44c340f45f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/research_projects/rag/eval_rag.py
View File

@@ -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,