"torchvision/vscode:/vscode.git/clone" did not exist on "3c33f367024a13596f94d816369ba142920839af"
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): ...@@ -197,13 +197,8 @@ 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