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

Commit d7ea4d5

Browse files
authored
Merge pull request #232 from stefan-it/update-readme
doc: update problem name for wmt task
2 parents 8f3a7fd + fed8cd7 commit d7ea4d5

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ trains an English-German translation model, and lets you use it interactively:
3434
pip install tensor2tensor && t2t-trainer \
3535
--generate_data \
3636
--data_dir=~/t2t_data \
37-
--problems=wmt_ende_tokens_32k \
37+
--problems=translate_ende_wmt32k \
3838
--model=transformer \
3939
--hparams_set=transformer_base_single_gpu \
4040
--output_dir=~/t2t_train/base \
@@ -72,7 +72,7 @@ pip install tensor2tensor
7272
# You can easily swap between them (and add new ones).
7373
t2t-trainer --registry_help
7474
75-
PROBLEM=wmt_ende_tokens_32k
75+
PROBLEM=translate_ende_wmt32k
7676
MODEL=transformer
7777
HPARAMS=transformer_base_single_gpu
7878
@@ -277,7 +277,7 @@ registrations.
277277
To add a new dataset, subclass
278278
[`Problem`](https://github.com/tensorflow/tensor2tensor/tree/master/tensor2tensor/data_generators/problem.py)
279279
and register it with `@registry.register_problem`. See
280-
[`WMTEnDeTokens8k`](https://github.com/tensorflow/tensor2tensor/tree/master/tensor2tensor/data_generators/wmt.py)
280+
[`TranslateEndeWmt8k`](https://github.com/tensorflow/tensor2tensor/tree/master/tensor2tensor/data_generators/wmt.py)
281281
for an example.
282282

283283
Also see the [data generators

0 commit comments

Comments
 (0)