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

Commit 1b6905f

Browse files
twairballrsepassi
authored andcommitted
fix #252 encoded vocab filepath for wmt17.zh-en (#293)
1 parent c3dedef commit 1b6905f

File tree

1 file changed

+2
-2
lines changed
  • tensor2tensor/data_generators

1 file changed

+2
-2
lines changed

tensor2tensor/data_generators/wmt.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -510,9 +510,9 @@ def target_space_id(self):
510510
def feature_encoders(self, data_dir):
511511
vocab_size = self.targeted_vocab_size
512512
source_vocab_filename = os.path.join(data_dir,
513-
"vocab.zh.%d" % vocab_size)
513+
"vocab.zhen-zh.%d" % vocab_size)
514514
target_vocab_filename = os.path.join(data_dir,
515-
"vocab.en.%d" % vocab_size)
515+
"vocab.zhen-en.%d" % vocab_size)
516516
source_token = text_encoder.SubwordTextEncoder(source_vocab_filename)
517517
target_token = text_encoder.SubwordTextEncoder(target_vocab_filename)
518518
return {

0 commit comments

Comments
 (0)