Skip to content

GitLab

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

Fixing learning_rate to consider all workers

parent 943dff84
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tutorials/image/cifar10_estimator/cifar10_main.py
View file @ f24c44d5
...@@ -226,7 +226,7 @@ def _resnet_model_fn(features, labels, mode): ...@@ -226,7 +226,7 @@ def _resnet_model_fn(features, labels, mode):
# https://github.com/ppwwyyxx/tensorpack/blob/master/examples/ResNet/cifar10-resnet.py#L155 # https://github.com/ppwwyyxx/tensorpack/blob/master/examples/ResNet/cifar10-resnet.py#L155
# users could apply other scheduling. # users could apply other scheduling.
num_batches_per_epoch = cifar10.Cifar10DataSet.num_examples_per_epoch( num_batches_per_epoch = cifar10.Cifar10DataSet.num_examples_per_epoch(
'train') // FLAGS.train_batch_size 'train') // (FLAGS.train_batch_size * FLAGS.num_workers)
boundaries = [ boundaries = [
num_batches_per_epoch * x num_batches_per_epoch * x
for x in np.array([82, 123, 300], dtype=np.int64) for x in np.array([82, 123, 300], dtype=np.int64)
......
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