diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-05 17:13:24 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-05 17:17:21 -0700 |
commit | 017599d0a1fa7a7227a43649db67e96311033a4e (patch) | |
tree | dc4b00270acc53b2e098022fb03e4b54cb40ab9e /tensorflow/python/BUILD | |
parent | e7b37766f53d5d9d976f2ba3046d3df3333c8ebb (diff) |
This CL changes the graph-mode API of the learning_rate_decay functions in TF 2.0 to return a no-arg callable to output a learning rate, instead of directly outputting a learning rate tensor.
This brings the graph mode API in line with the eager execution API, where this change was made to allow changing the learning rate value across different invocations of optimizer functions.
PiperOrigin-RevId: 211726295
Diffstat (limited to 'tensorflow/python/BUILD')
-rw-r--r-- | tensorflow/python/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index e6169e9e80..ba9c6a2320 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -4393,6 +4393,7 @@ cuda_py_tests( "training/ftrl_test.py", "training/gradient_descent_test.py", "training/learning_rate_decay_test.py", + "training/learning_rate_decay_v2_test.py", "training/momentum_test.py", "training/optimizer_test.py", "training/proximal_adagrad_test.py", |