Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "4f36084ff7f55a59407b8f82cabfddb9f9215517"
Unverified Commit 649e0374 authored by Amit Patankar's avatar Amit Patankar Committed by GitHub
Browse files

Updating the learning_rate_fn for keras cifar.

parent b2af31a9
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
official/resnet/cifar10_main.py
View file @ 649e0374
...@@ -197,8 +197,13 @@ def cifar10_model_fn(features, labels, mode, params): ...@@ -197,8 +197,13 @@ def cifar10_model_fn(features, labels, mode, params):
"""Model function for CIFAR-10.""" """Model function for CIFAR-10."""
features = tf.reshape(features, [-1, _HEIGHT, _WIDTH, _NUM_CHANNELS]) features = tf.reshape(features, [-1, _HEIGHT, _WIDTH, _NUM_CHANNELS])
# Learning rate schedule follows arXiv:1512.03385 for ResNet-56 and under. # 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( learning_rate_fn = resnet_run_loop.learning_rate_with_decay(
batch_size=params['batch_size'], batch_denom=128, 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], num_images=_NUM_IMAGES['train'], boundary_epochs=[91, 136, 182],
decay_rates=[1, 0.1, 0.01, 0.001]) 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