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

[s2s] round bleu, rouge to 4 digits (#6704)

Browse Source
This commit is contained in:
Sam Shleifer
2020-08-25 00:33:11 -04:00
committed by GitHub
parent b6512d2357
commit 0344428f79
4 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/seq2seq/distillation.py
View File

@@ -20,7 +20,7 @@ try:
from .utils import (
any_requires_grad,
assert_all_frozen,
calculate_bleu_score,
calculate_bleu,
freeze_params,
pickle_load,
use_task_specific_params,
@@ -32,7 +32,7 @@ except ImportError:
from utils import (
any_requires_grad,
assert_all_frozen,
calculate_bleu_score,
calculate_bleu,
freeze_params,
pickle_load,
use_task_specific_params,
@@ -261,7 +261,7 @@ class BartTranslationDistiller(BartSummarizationDistiller):
self.decoder_start_token_id = self.tokenizer.lang_code_to_id[hparams.tgt_lang]
def calc_generative_metrics(self, preds, target) -> dict:
return calculate_bleu_score(preds, target)
return calculate_bleu(preds, target)
@staticmethod
def add_model_specific_args(parser, root_dir):