Commit 147e6777 authored by Allen Lavoie's avatar Allen Lavoie Committed by TensorFlower Gardener
Browse files

Fix checkpoint compatibility issues for optimizers in SavedModels 

Does not yet revive fully functional optimizer objects. Right now the revived object is just a place to put slot variables which are referenced elsewhere. But at least objects which reference optimizers are now saveable and restorable, and traced functions which use the optimizer work fine.

PiperOrigin-RevId: 238660860
parent 9c6d4589
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