addons icon indicating copy to clipboard operation
addons copied to clipboard

`MultiOptimizer` is not serializable.

Open JackWindows opened this issue 2 years ago • 0 comments

The current implementation of MultiOptimizer puts grads and vars inside the optimizer_specs variable. Since the grads and vars are not serializable, any model that uses MultiOptimizer class cannot be saved or checkpointed. This PR will only extract the necessary optimizer_specs components that are useful for re-instantiating the MultiOptimizer instance inside get_config() method, leading to a serializable MultiOptimizer implementation.

https://github.com/tensorflow/addons/pull/2719

JackWindows avatar Jun 24 '22 18:06 JackWindows