Commit aee80c4b authored by Sergio Guadarrama's avatar Sergio Guadarrama Committed by TensorFlower Gardener
Browse files

Make sure that variables.trainable and considered trainable. 

When creating models under tf.function the trainable variables don't always end up in tf.trainable_variables() collection.

PiperOrigin-RevId: 232531430
parent 1b24197e
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