tf.compat.v1.train.update_checkpoint_state