Commit b76620ae authored by James Keeling's avatar James Keeling Committed by TensorFlower Gardener
Browse files

Default to previously specified variables when minimizing with KfacOptimizer 

If no variables are specified to minimize or compute_gradients, the default was previously to use all trainable variables. However, KfacOptimizer has a list of variables it is able to train, so we should use that instead.

PiperOrigin-RevId: 176720954
parent 4e5534d3
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