Correct the spelling of bleu metric (#18375)

This commit is contained in:
Ogundepo Odunayo 2022-08-01 07:51:27 -04:00 committed by GitHub
parent 1f84399171
commit 679d68a11b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions

View File

@ -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"}

View File

@ -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,