Skip to content

GitLab

Commit 77cab72c authored by Neal Wu's avatar Neal Wu
Browse files

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

parent ecf24560
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tutorials/rnn/ptb/ptb_word_lm.py
View file @ 77cab72c
...@@ -331,7 +331,7 @@ class SmallConfig(object): ...@@ -331,7 +331,7 @@ class SmallConfig(object):
lr_decay = 0.5 lr_decay = 0.5
batch_size = 20 batch_size = 20
vocab_size = 10000 vocab_size = 10000
rnn_mode = CUDNN rnn_mode = BLOCK
class MediumConfig(object): class MediumConfig(object):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment