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

style fix

parent b98bee19
...@@ -318,8 +318,7 @@ def _resnet_model_fn(features, labels, mode): ...@@ -318,8 +318,7 @@ def _resnet_model_fn(features, labels, mode):
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 FLAGS.learning_rate * x for x in [1, 0.1, 0.01, 0.002]
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(),
......
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