Change the default rnn_mode for ptb_word_lm.py to BLOCK so that it wo…

…rks with TF1.4
cshuangyi82 · Nov 29, 2017 · 77cab72 · 77cab72
1 parent ecf2456
commit 77cab72
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):