Prepare checkpoint variables in a separate graph
@@ -79,6 +79,7 @@ def _create_checkpoint_from_variables(variables, output_dir, latest_step=None, s
if "global_step" in variables:
latest_step = variables["global_step"]
del variables["global_step"]
+ with tf.Graph().as_default():
tf_vars = [
tf.get_variable(
name,
