From ad98642a82a4f157b05d7f4af25bec1ae74077ba Mon Sep 17 00:00:00 2001 From: Zachary Mueller Date: Wed, 5 Oct 2022 14:52:01 -0400 Subject: [PATCH] Fix gather for metrics (#19360) --- examples/pytorch/summarization/run_summarization_no_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/pytorch/summarization/run_summarization_no_trainer.py b/examples/pytorch/summarization/run_summarization_no_trainer.py index 594d468330..9b8ee1a1dc 100644 --- a/examples/pytorch/summarization/run_summarization_no_trainer.py +++ b/examples/pytorch/summarization/run_summarization_no_trainer.py @@ -685,7 +685,7 @@ def main(): # If we did not pad to max length, we need to pad the labels too labels = accelerator.pad_across_processes(batch["labels"], dim=1, pad_index=tokenizer.pad_token_id) - generated_tokens, labels = accelerator.gather_for_metrics(generated_tokens, labels) + generated_tokens, labels = accelerator.gather_for_metrics((generated_tokens, labels)) generated_tokens = generated_tokens.cpu().numpy() labels = labels.cpu().numpy()