diff options
Diffstat (limited to 'Code/avg_runner.py')
| -rw-r--r-- | Code/avg_runner.py | 2 |
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: |