Commit 1d0ec3ec authored by Katherine Wu's avatar Katherine Wu Committed by TensorFlower Gardener
Browse files

Enable export of train/eval graphs for Keras models that use OptimizerV2. 

Since OptimizerV2's get_config requires using the backend to get the values of hyperparameters, some code had to be moved into the Session that is created during export.

PiperOrigin-RevId: 235039100
parent b78cce68
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