summaryrefslogtreecommitdiff
path: root/Code/avg_runner.py
diff options
context:
space:
mode:
authorMatt Cooper <matthew_cooper@brown.edu>2016-12-09 18:47:49 -0500
committerMatt Cooper <matthew_cooper@brown.edu>2016-12-09 18:47:49 -0500
commit0197b738067ed76cba63b0eff41ca65ae5e0341d (patch)
tree4f03b188a35d43942cb5dd01b53fc043ee90c2ff /Code/avg_runner.py
parent9c6b43a967227f7bea591f1647b8dff9953cd7ad (diff)
Updated for tensorflow 0.12
Diffstat (limited to 'Code/avg_runner.py')
-rw-r--r--Code/avg_runner.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/Code/avg_runner.py b/Code/avg_runner.py
index 959ff90..6809187 100644
--- a/Code/avg_runner.py
+++ b/Code/avg_runner.py
@@ -51,7 +51,7 @@ class AVGRunner:
print 'Init variables...'
self.saver = tf.train.Saver(keep_checkpoint_every_n_hours=2)
- self.sess.run(tf.initialize_all_variables())
+ self.sess.run(tf.global_variables_initializer())
# if load path specified, load a saved model
if model_load_path is not None: