"docs/vscode:/vscode.git/clone" did not exist on "b70e333ac5be307d2d5e38c1653b755907a09f04"
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
......@@ -226,7 +226,7 @@ def _resnet_model_fn(features, labels, mode):
# https://github.com/ppwwyyxx/tensorpack/blob/master/examples/ResNet/cifar10-resnet.py#L155
# users could apply other scheduling.
num_batches_per_epoch = cifar10.Cifar10DataSet.num_examples_per_epoch(
'train') // FLAGS.train_batch_size
'train') // (FLAGS.train_batch_size * FLAGS.num_workers)
boundaries = [
num_batches_per_epoch * x
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