Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "b3595cf02b9997e9acfc23ed6f4bcec78cd315ed"
Unverified Commit 05b4c329 authored by Bhadresh Savani's avatar Bhadresh Savani Committed by GitHub
Browse files

fixed a typo (#16508)

parent 6a4dbba1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/question-answering/run_qa_no_trainer.py
View file @ 05b4c329
...@@ -791,7 +791,7 @@ def main(): ...@@ -791,7 +791,7 @@ def main():
if not args.pad_to_max_length: # necessary to pad predictions and labels for being gathered if not args.pad_to_max_length: # necessary to pad predictions and labels for being gathered
start_logits = accelerator.pad_across_processes(start_logits, dim=1, pad_index=-100) start_logits = accelerator.pad_across_processes(start_logits, dim=1, pad_index=-100)
end_logits = accelerator.pad_across_processes(start_logits, dim=1, pad_index=-100) end_logits = accelerator.pad_across_processes(end_logits, dim=1, pad_index=-100)
all_start_logits.append(accelerator.gather(start_logits).cpu().numpy()) all_start_logits.append(accelerator.gather(start_logits).cpu().numpy())
all_end_logits.append(accelerator.gather(end_logits).cpu().numpy()) all_end_logits.append(accelerator.gather(end_logits).cpu().numpy())
......
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