Unverified Commit 679d68a1 authored by Ogundepo Odunayo's avatar Ogundepo Odunayo Committed by GitHub
Browse files

Correct the spelling of bleu metric (#18375)

parent 1f843991
......@@ -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"}
......
......@@ -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,
......
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