ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


View source on GitHub

Executes training steps, recovers and checkpoints.

Note that this class is particularly preliminary, experimental, and expected to change.

step_callable a training Step that's capable of signaling when done.
session a Session instance that's needed for graph mode.

ValueError if session was provided for eager mode or not provided for graph mode.



View source