From 679d68a11be278b798e468bacafa710b544f807a Mon Sep 17 00:00:00 2001 From: Ogundepo Odunayo Date: Mon, 1 Aug 2022 07:51:27 -0400 Subject: [PATCH] Correct the spelling of bleu metric (#18375) --- examples/pytorch/translation/run_translation.py | 4 ++-- examples/pytorch/translation/run_translation_no_trainer.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/pytorch/translation/run_translation.py b/examples/pytorch/translation/run_translation.py index 4d03cbf49c..a519fa1753 100755 --- a/examples/pytorch/translation/run_translation.py +++ b/examples/pytorch/translation/run_translation.py @@ -119,12 +119,12 @@ class DataTrainingArguments: validation_file: Optional[str] = field( default=None, metadata={ - "help": "An optional input evaluation data file to evaluate the metrics (sacreblue) on a jsonlines file." + "help": "An optional input evaluation data file to evaluate the metrics (sacrebleu) on a jsonlines file." }, ) test_file: Optional[str] = field( default=None, - metadata={"help": "An optional input test data file to evaluate the metrics (sacreblue) on a jsonlines file."}, + metadata={"help": "An optional input test data file to evaluate the metrics (sacrebleu) on a jsonlines file."}, ) overwrite_cache: bool = field( default=False, metadata={"help": "Overwrite the cached training and evaluation sets"} diff --git a/examples/pytorch/translation/run_translation_no_trainer.py b/examples/pytorch/translation/run_translation_no_trainer.py index 914181511b..0054526f1e 100644 --- a/examples/pytorch/translation/run_translation_no_trainer.py +++ b/examples/pytorch/translation/run_translation_no_trainer.py @@ -695,7 +695,7 @@ def main(): if args.with_tracking: accelerator.log( { - "blue": eval_metric["score"], + "bleu": eval_metric["score"], "train_loss": total_loss.item() / len(train_dataloader), "epoch": epoch, "step": completed_steps,