Searched refs:_StopAtSweepHook (Results 1 – 2 of 2) sorted by relevance
111 class _StopAtSweepHook(session_run_hook.SessionRunHook): class307 training_hooks.append(_StopAtSweepHook(max_sweeps))
488 hook = wals_lib._StopAtSweepHook(last_sweep=10)