Skip to content

GitLab

Commit 5ad0c9e8 authored by Frederick Liu's avatar Frederick Liu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 352011767
parent 413d3f7e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/nlp/tasks/translation.py
View file @ 5ad0c9e8
...@@ -28,8 +28,8 @@ from official.core import config_definitions as cfg ...@@ -28,8 +28,8 @@ from official.core import config_definitions as cfg
from official.core import task_factory from official.core import task_factory
from official.modeling.hyperparams import base_config from official.modeling.hyperparams import base_config
from official.nlp.data import data_loader_factory from official.nlp.data import data_loader_factory
from official.nlp.metrics import bleu
from official.nlp.modeling import models from official.nlp.modeling import models
from official.nlp.transformer import compute_bleu
def _pad_tensors_to_same_length(x, y): def _pad_tensors_to_same_length(x, y):
...@@ -364,6 +364,6 @@ class TranslationTask(base_task.Task): ...@@ -364,6 +364,6 @@ class TranslationTask(base_task.Task):
src, translation, self._references[u_id]) src, translation, self._references[u_id])
sacrebleu_score = sacrebleu.corpus_bleu( sacrebleu_score = sacrebleu.corpus_bleu(
translations, [self._references]).score translations, [self._references]).score
bleu_score = compute_bleu.bleu_on_list(self._references, translations) bleu_score = bleu.bleu_on_list(self._references, translations)
return {"sacrebleu_score": sacrebleu_score, return {"sacrebleu_score": sacrebleu_score,
"bleu_score": bleu_score} "bleu_score": bleu_score}
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