Commit 5f1a66cc authored by Igor Saprykin's avatar Igor Saprykin Committed by TensorFlower Gardener
Browse files

Add more recovery functionality to MonitoredSession.run_step_fn. 

Current implemention wouldn't recover from one of `_PREEMPTION_ERRORS` during a fetch through the raw session that is made available to the step_fn.

The changelist presents a way to map the desired functionality to the hiearchy of _MonitoredSession > (possibly!) _RecoverableSession > _CoordinatedSession > _HookedSession.

PiperOrigin-RevId: 174053865
parent b2ff3ad9
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment