tf.estimator.SessionRunHook