Skip to content

Commit

Permalink
fix prediction error:tensorflow/nmt#341
Browse files Browse the repository at this point in the history
  • Loading branch information
ChinaShrimp committed Jun 25, 2018
1 parent 27e7077 commit 0d44954
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions gnmt/wmt16.json.diff
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
diff --git a/nmt/standard_hparams/wmt16.json b/nmt/standard_hparams/wmt16.json
index 8c1cb3f..afa38f9 100644
--- a/nmt/standard_hparams/wmt16.json
+++ b/nmt/standard_hparams/wmt16.json
@@ -14,6 +14,8 @@
"metrics": ["bleu"],
"num_buckets": 5,
"num_layers": 4,
+ "num_encoder_layers": 4,
+ "num_decoder_layers": 4,
"num_train_steps": 340000,
"decay_scheme": "luong10",
"num_units": 1024,

0 comments on commit 0d44954

Please sign in to comment.