Monitor to Request Stop at a Specified Step
hook_stop_at_step(num_steps = NULL, last_step = NULL)Number of steps to execute.
Step after which to stop.
Other session_run_hook wrappers:
hook_checkpoint_saver(),
hook_global_step_waiter(),
hook_history_saver(),
hook_logging_tensor(),
hook_nan_tensor(),
hook_progress_bar(),
hook_step_counter(),
hook_summary_saver(),
session_run_hook()