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: ...@@ -119,12 +119,12 @@ class DataTrainingArguments:
validation_file: Optional[str] = field( validation_file: Optional[str] = field(
default=None, default=None,
metadata={ 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( test_file: Optional[str] = field(
default=None, 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( overwrite_cache: bool = field(
default=False, metadata={"help": "Overwrite the cached training and evaluation sets"} default=False, metadata={"help": "Overwrite the cached training and evaluation sets"}
......
...@@ -695,7 +695,7 @@ def main(): ...@@ -695,7 +695,7 @@ def main():
if args.with_tracking: if args.with_tracking:
accelerator.log( accelerator.log(
{ {
"blue": eval_metric["score"], "bleu": eval_metric["score"],
"train_loss": total_loss.item() / len(train_dataloader), "train_loss": total_loss.item() / len(train_dataloader),
"epoch": epoch, "epoch": epoch,
"step": completed_steps, "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