tf.estimator.StopAtStepHook