class_docstr: Load checkpoint right after the session started.
class_docstr_tok: ["Load","checkpoint","right","after","the","session","started","."]
code: def after_create_session(self, session, coord):
      # pylint: disable=protected-access
      saver_collection = tf.get_collection(tf.GraphKeys.SAVERS)
      if saver_collection:
          saver = saver_collection[0]
          check_point = tf.train.latest_checkpoint(FLAGS.checkpoint_dir)
          if check_point:
              saver.restore(session, check_point)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import re
  import mesh_tensorflow
  import numpy
  import six
  from six.moves import range
  from six.moves import zip
  import tensorflow.compat.v1
  from tensorflow.compat.v1 import estimator
  from mesh_tensorflow.experimental import input_reader
  from mesh_tensorflow.experimental import unet
  from tensorflow.contrib import summary
  from tensorflow.contrib import tpu
  from tensorflow.contrib.tpu.python.tpu import device_assignment
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import control_flow_ops
  from tensorflow.python.platform import flags
  from tensorflow.python.tpu.ops import tpu_ops
label_desc: Use image node_id 1 for calling the _CkptLoaderHook obj's underlying member method code with example usage: obj.after_create_session(session, coord) without return types