def after_run(self, run_context: tf.estimator.SessionRunContext, run_values: tf.estimator.SessionRunValues) -> None: if run_context.stop_requested: run_context._stop_requested = False self.context.set_stop_requested(True)
def after_run(self, run_context: tf.estimator.SessionRunContext, run_values: tf.estimator.SessionRunValues) -> None: run_context.request_stop()