Fix deprecated functions warnings

Beokane · Dec 5, 2016 · 673b7d1 · 673b7d1
1 parent c008eba
commit 673b7d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -126,13 +126,13 @@
         checkpoint_prefix = os.path.join(checkpoint_dir, "model")
         if not os.path.exists(checkpoint_dir):
             os.makedirs(checkpoint_dir)
-        saver = tf.train.Saver(tf.all_variables())
+        saver = tf.train.Saver(tf.global_variables())
 
         # Write vocabulary
         vocab_processor.save(os.path.join(out_dir, "vocab"))
 
         # Initialize all variables
-        sess.run(tf.initialize_all_variables())
+        sess.run(tf.global_variables_initializer())
 
         def train_step(x_batch, y_batch):
             """