"git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "8cddfa56a14e9ac03677bfc9e8df2f59b5bce382"
Commit abdc7f11 authored by Amit Patankar's avatar Amit Patankar
Browse files

Revert "Updating the learning_rate_fn for keras cifar."

This reverts commit 649e0374.
parent 649e0374
......@@ -197,13 +197,8 @@ def cifar10_model_fn(features, labels, mode, params):
"""Model function for CIFAR-10."""
features = tf.reshape(features, [-1, _HEIGHT, _WIDTH, _NUM_CHANNELS])
# Learning rate schedule follows arXiv:1512.03385 for ResNet-56 and under.
#learning_rate_fn = resnet_run_loop.learning_rate_with_decay(
# batch_size=params['batch_size'], batch_denom=128,
# num_images=_NUM_IMAGES['train'], boundary_epochs=[91, 136, 182],
# decay_rates=[1, 0.1, 0.01, 0.001])
learning_rate_fn = resnet_run_loop.learning_rate_with_decay(
batch_size=params['batch_size'], batch_denom=128,
num_images=_NUM_IMAGES['train'], boundary_epochs=[100, 150, 200],
num_images=_NUM_IMAGES['train'], boundary_epochs=[91, 136, 182],
decay_rates=[1, 0.1, 0.01, 0.001])
......
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