Switch to the new FileWriter API

2af8c3d2 · Neal Wu · f7fd0012 · 2af8c3d2
Commit 2af8c3d2 authored Mar 14, 2017 by Neal Wu
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

inception/inception/inception_train.py inception/inception/inception_train.py +1 -1

No files found.
--- a/inception/inception/inception_train.py
+++ b/inception/inception/inception_train.py
@@ -331,7 +331,7 @@ def train(dataset):

    summary_writer = tf.summary.FileWriter(
        FLAGS.train_dir,
-        graph_def=sess.graph.as_graph_def(add_shapes=True))
+        graph=sess.graph)

    for step in range(FLAGS.max_steps):
      start_time = time.time()