Python estimator.SessionRunContext Examples

Example #1

 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)

Example #2

File: estimator_xor_model.py Project: justin-determined-ai/determined

 def after_run(self, run_context: tf.estimator.SessionRunContext,
               run_values: tf.estimator.SessionRunValues) -> None:
     run_context.request_stop()