tf.estimator.experimental.InMemoryEvaluatorHook