mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-03 03:31:05 +06:00
Correct the spelling of bleu metric (#18375)
This commit is contained in:
parent
1f84399171
commit
679d68a11b
@ -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,
|
||||
|
Loading…
Reference in New Issue
Block a user