diff options
| author |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-21 12:25:55 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-21 12:25:55 -0700 |
| commit | 237cb7f7d0b0fd078d6482a55f3733f51b1ad09e (patch) | |
| tree | ab80d4bc135eca47ce1e5fab53e226678cbd8287 /tensorflow/contrib/opt | |
| parent | 041b02c3b48a9b0cb085ca02f8674171f5d31fca (diff) | |
| parent | ac15dfe64ea7c122f7609ae5f3f927447b41a02e (diff) | |
Merge pull request #18959 from Rholais:patch-1
PiperOrigin-RevId: 209638931
Diffstat (limited to 'tensorflow/contrib/opt')
| -rw-r--r-- | tensorflow/contrib/opt/python/training/powersign.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/opt/python/training/powersign.py b/tensorflow/contrib/opt/python/training/powersign.py index 828f3c51c9..b4aa19264d 100644 --- a/tensorflow/contrib/opt/python/training/powersign.py +++ b/tensorflow/contrib/opt/python/training/powersign.py @@ -65,7 +65,7 @@ class PowerSignOptimizer(optimizer.Optimizer): Example usage for PowerSign-cd (PowerSign with cosine sign decay) ``` decay_steps = 1000 - linear_decay_fn = sign_decays.get_linear_decay_fn(decay_steps) + linear_decay_fn = sign_decays.get_cosine_decay_fn(decay_steps) opt = PowerSignOptimizer(learning_rate=0.1, sign_decay_fn=linear_decay_fn) ``` |