Fix init_fn in train_utils. (#7790)

87bc52c5 · Yukun Zhu · aquariusjay · d48d036a · 87bc52c5
Commit 87bc52c5 authored Nov 11, 2019 by Yukun Zhu Committed by aquariusjay Nov 11, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

research/deeplab/utils/train_utils.py research/deeplab/utils/train_utils.py +1 -1

No files found.
--- a/research/deeplab/utils/train_utils.py
+++ b/research/deeplab/utils/train_utils.py
@@ -221,7 +221,7 @@ def get_model_init_fn(train_logdir,
        ignore_missing_vars=ignore_missing_vars)
    global_step = tf.train.get_or_create_global_step()

-    def restore_fn(unused_scaffold, sess):
+    def restore_fn(sess):
      sess.run(init_op, init_feed_dict)
      sess.run([global_step])