diff options
author | 2016-02-23 18:51:19 -0800 | |
---|---|---|
committer | 2016-02-24 15:32:39 -0800 | |
commit | 94a992cfc3266244b81fe311b805fc1ae3f53f30 (patch) | |
tree | 0a09c28a42de2ab4c6ce1e333c258fb7e5863fe0 /tensorflow/models/image | |
parent | 185cff7f41600e26d5f9ecfb18bcaab09edf4bfe (diff) |
Add correct dependencies to sdca ops to fix build breakage.
Change: 115408162
Diffstat (limited to 'tensorflow/models/image')
-rw-r--r-- | tensorflow/models/image/cifar10/cifar10_eval.py | 2 | ||||
-rw-r--r-- | tensorflow/models/image/cifar10/cifar10_multi_gpu_train.py | 3 | ||||
-rw-r--r-- | tensorflow/models/image/cifar10/cifar10_train.py | 3 |
3 files changed, 5 insertions, 3 deletions
diff --git a/tensorflow/models/image/cifar10/cifar10_eval.py b/tensorflow/models/image/cifar10/cifar10_eval.py index 5395dbc6ca..57eb94c3c9 100644 --- a/tensorflow/models/image/cifar10/cifar10_eval.py +++ b/tensorflow/models/image/cifar10/cifar10_eval.py @@ -136,7 +136,7 @@ def evaluate(): # Build the summary operation based on the TF collection of Summaries. summary_op = tf.merge_all_summaries() - graph_def = tf.get_default_graph().as_graph_def() + graph_def = tf.get_default_graph().as_graph_def(add_shapes=True) summary_writer = tf.train.SummaryWriter(FLAGS.eval_dir, graph_def=graph_def) diff --git a/tensorflow/models/image/cifar10/cifar10_multi_gpu_train.py b/tensorflow/models/image/cifar10/cifar10_multi_gpu_train.py index 0846ab6d21..3e3d21cb53 100644 --- a/tensorflow/models/image/cifar10/cifar10_multi_gpu_train.py +++ b/tensorflow/models/image/cifar10/cifar10_multi_gpu_train.py @@ -239,8 +239,9 @@ def train(): # Start the queue runners. tf.train.start_queue_runners(sess=sess) + graph_def = sess.graph.as_graph_def(add_shapes=True) summary_writer = tf.train.SummaryWriter(FLAGS.train_dir, - graph_def=sess.graph_def) + graph_def=graph_def) for step in xrange(FLAGS.max_steps): start_time = time.time() diff --git a/tensorflow/models/image/cifar10/cifar10_train.py b/tensorflow/models/image/cifar10/cifar10_train.py index dc9d0b623a..cf59894a81 100644 --- a/tensorflow/models/image/cifar10/cifar10_train.py +++ b/tensorflow/models/image/cifar10/cifar10_train.py @@ -93,8 +93,9 @@ def train(): # Start the queue runners. tf.train.start_queue_runners(sess=sess) + graph_def = sess.graph.as_graph_def(add_shapes=True) summary_writer = tf.train.SummaryWriter(FLAGS.train_dir, - graph_def=sess.graph_def) + graph_def=graph_def) for step in xrange(FLAGS.max_steps): start_time = time.time() |