update

ace19-dev · Aug 9, 2019 · d8cd0cc · d8cd0cc
1 parent f7a52f5
commit d8cd0cc
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/utils/train_utils.py b/utils/train_utils.py
@@ -386,8 +386,7 @@ def custom_restore_fn(flags):
                                  flags.checkpoint_model_scope + var.op.name): var
              for var in variables_to_restore}
 
-    slim.assign_from_checkpoint_fn(flags.checkpoint_dir,
-                                   variables_to_restore)
+    slim.assign_from_checkpoint_fn(flags.checkpoint_dir, variables_to_restore)
     tf.compat.v1.logging.info('Fine-tuning from %s.' % flags.checkpoint_dir)