Merge pull request #2920 from tensorflow/change-ptb-defaults

Change the default rnn_mode for ptb_word_lm.py to BLOCK so that it works with TF1.4

Merge pull request #2920 from tensorflow/change-ptb-defaults
Change the default rnn_mode for ptb_word_lm.py to BLOCK so that it works with TF1.4
42a3da72 · Neal Wu · GitHub · ecf24560 · 77cab72c · 42a3da72
Unverified Commit 42a3da72 authored Nov 28, 2017 by Neal Wu Committed by GitHub Nov 28, 2017
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tutorials/rnn/ptb/ptb_word_lm.py tutorials/rnn/ptb/ptb_word_lm.py +1 -1

No files found.
--- a/tutorials/rnn/ptb/ptb_word_lm.py
+++ b/tutorials/rnn/ptb/ptb_word_lm.py
@@ -331,7 +331,7 @@ class SmallConfig(object):
  lr_decay = 0.5
  batch_size = 20
  vocab_size = 10000
-  rnn_mode = CUDNN
+  rnn_mode = BLOCK


 class MediumConfig(object):