Skip to content
This repository was archived by the owner on Jul 7, 2023. It is now read-only.

Commit b8aa4ea

Browse files
nshazeerlukaszkaiser
authored and
lukaszkaiser
committed
add translate_enfr_wmt32k_packed problem.
PiperOrigin-RevId: 177752030
1 parent 9a8f203 commit b8aa4ea

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

Diff for: tensor2tensor/data_generators/translate_enfr.py

+8
Original file line numberDiff line numberDiff line change
@@ -143,6 +143,14 @@ def use_small_dataset(self):
143143
return False
144144

145145

146+
@registry.register_problem
147+
class TranslateEnfrWmt32kPacked(TranslateEnfrWmt32k):
148+
149+
@property
150+
def packed_length(self):
151+
return 256
152+
153+
146154
@registry.register_problem
147155
class TranslateEnfrWmtSmallCharacters(translate.TranslateProblem):
148156
"""Problem spec for WMT En-Fr translation."""

0 commit comments

Comments
 (0)