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

Commit 3913c1e

Browse files
authored
Merge pull request #181 from martinpopel/add_czech2
Simplify adding new translation language pairs
2 parents 668e385 + f793de2 commit 3913c1e

File tree

3 files changed

+176
-135
lines changed

3 files changed

+176
-135
lines changed

tensor2tensor/data_generators/problem.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,10 @@ class SpaceID(object):
6969
ICE_PARSE_TOK = 19
7070
# Macedonian tokens
7171
MK_TOK = 20
72+
# Czech tokens
73+
CS_TOK = 21
74+
# Czech characters
75+
CS_CHR = 22
7276

7377

7478
class Problem(object):

tensor2tensor/data_generators/problem_hparams.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -180,6 +180,9 @@ def default_problem_hparams():
180180
# 17: Icelandic characters
181181
# 18: Icelandic tokens
182182
# 19: Icelandic parse tokens
183+
# 20: Macedonian tokens
184+
# 21: Czech tokens
185+
# 22: Czech characters
183186
# Add more above if needed.
184187
input_space_id=0,
185188
target_space_id=0,

0 commit comments

Comments
 (0)