Skip to content

GitLab

Commit 01122544 authored by Marianne Linhares Monteiro's avatar Marianne Linhares Monteiro Committed by GitHub
Browse files

staged_lr in a single line.

parent 24b80af4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
tutorials/image/cifar10_estimator/cifar10_main.py
View file @ 01122544
...@@ -317,9 +317,7 @@ def _resnet_model_fn(features, labels, mode): ...@@ -317,9 +317,7 @@ def _resnet_model_fn(features, labels, mode):
num_batches_per_epoch * x num_batches_per_epoch * x
for x in np.array([82, 123, 300], dtype=np.int64) for x in np.array([82, 123, 300], dtype=np.int64)
] ]
staged_lr = [ staged_lr = [FLAGS.learning_rate * x for x in [1, 0.1, 0.01, 0.002]]
FLAGS.learning_rate * x for x in [1, 0.1, 0.01, 0.002]
]
learning_rate = tf.train.piecewise_constant(tf.train.get_global_step(), learning_rate = tf.train.piecewise_constant(tf.train.get_global_step(),
boundaries, staged_lr) boundaries, staged_lr)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment