Commit 2e272dbc authored by Allen Lavoie's avatar Allen Lavoie Committed by TensorFlower Gardener
Browse files

Make the TFOptimizer wrapper checkpointable. 

TensorFlow Optimizers compiled with a Model will now have their state saved and restored with save_weights/load_weights.

PiperOrigin-RevId: 198811639
parent c7c95eee
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