SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Correct the spelling of bleu metric (#18375)

Browse Source
This commit is contained in:
Ogundepo Odunayo
2022-08-01 07:51:27 -04:00
committed by GitHub
parent 1f84399171
commit 679d68a11b
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/pytorch/translation/run_translation.py
View File

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

2
examples/pytorch/translation/run_translation_no_trainer.py
View File

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