Skip to content

GitLab

Commit a4ae812e authored by Chen Qian's avatar Chen Qian Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 471682920
parent 56453cd5
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 2 deletions
official/legacy/image_classification/resnet/common.py
View file @ a4ae812e
...@@ -106,9 +106,13 @@ class PiecewiseConstantDecayWithWarmup( ...@@ -106,9 +106,13 @@ class PiecewiseConstantDecayWithWarmup(
} }
def get_optimizer(learning_rate=0.1): def get_optimizer(learning_rate=0.1, use_legacy_optimizer=True):
"""Returns optimizer to use.""" """Returns optimizer to use."""
# The learning_rate is overwritten at the beginning of each step by callback. # The learning_rate is overwritten at the beginning of each step by callback.
if use_legacy_optimizer:
return tf.keras.optimizers.legacy.SGD(
learning_rate=learning_rate, momentum=0.9)
else:
return tf.keras.optimizers.SGD(learning_rate=learning_rate, momentum=0.9) return tf.keras.optimizers.SGD(learning_rate=learning_rate, momentum=0.9)
......
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