Merge pull request #2259 from skwbc/str_decode_py3

bugfix for rnn tutorial on Python3

Merge pull request #2259 from skwbc/str_decode_py3
bugfix for rnn tutorial on Python3
bc86a539 · Toby Boyd · GitHub · 55e6e387 · 812bc2e7 · bc86a539
Commit bc86a539 authored Sep 08, 2017 by Toby Boyd Committed by GitHub Sep 08, 2017
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

tutorials/rnn/ptb/reader.py tutorials/rnn/ptb/reader.py +5 -1

No files found.
--- a/tutorials/rnn/ptb/reader.py
+++ b/tutorials/rnn/ptb/reader.py
@@ -25,9 +25,13 @@ import sys

 import tensorflow as tf

+Py3 = sys.version_info[0] == 3

 def _read_words(filename):
  with tf.gfile.GFile(filename, "r") as f:
+    if Py3:
+      return f.read().replace("\n", "<eos>").split()
+    else:
      return f.read().decode("utf-8").replace("\n", "<eos>").split()