diff options
author | Dan Mané <danmane@google.com> | 2016-11-29 16:48:26 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-11-29 17:09:16 -0800 |
commit | 649095a159cf8dadeb49291bd5ad6fe6d8f52740 (patch) | |
tree | c546ae1b35c340b0f731e4ad5219dde58cdd7e97 /tensorflow/examples/image_retraining | |
parent | 7c4333c3f4aac04140739cbd527247a7e07de014 (diff) |
Replace tf.train.SummaryWriter with tf.summary.FileWriter
Change: 140545243
Diffstat (limited to 'tensorflow/examples/image_retraining')
-rw-r--r-- | tensorflow/examples/image_retraining/retrain.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/examples/image_retraining/retrain.py b/tensorflow/examples/image_retraining/retrain.py index 9e2d01615d..ca8c9358b3 100644 --- a/tensorflow/examples/image_retraining/retrain.py +++ b/tensorflow/examples/image_retraining/retrain.py @@ -810,9 +810,9 @@ def main(_): # Merge all the summaries and write them out to /tmp/retrain_logs (by default) merged = tf.summary.merge_all() - train_writer = tf.train.SummaryWriter(FLAGS.summaries_dir + '/train', - sess.graph) - validation_writer = tf.train.SummaryWriter(FLAGS.summaries_dir + '/validation') + train_writer = tf.summary.FileWriter(FLAGS.summaries_dir + '/train', + sess.graph) + validation_writer = tf.summary.FileWriter(FLAGS.summaries_dir + '/validation') # Set up all our weights to their initial default values. init = tf.global_variables_initializer() |