Unverified Commit c4c49d1a authored by Jon Shlens's avatar Jon Shlens Committed by GitHub
Browse files

Merge pull request #5658 from cheerss/master

update the calculation of num_batches_per_epoch
parents 82e783e3 c8032908
...@@ -147,7 +147,7 @@ def train(): ...@@ -147,7 +147,7 @@ def train():
# Calculate the learning rate schedule. # Calculate the learning rate schedule.
num_batches_per_epoch = (cifar10.NUM_EXAMPLES_PER_EPOCH_FOR_TRAIN / num_batches_per_epoch = (cifar10.NUM_EXAMPLES_PER_EPOCH_FOR_TRAIN /
FLAGS.batch_size) FLAGS.batch_size / FLAGS.num_gpus)
decay_steps = int(num_batches_per_epoch * cifar10.NUM_EPOCHS_PER_DECAY) decay_steps = int(num_batches_per_epoch * cifar10.NUM_EPOCHS_PER_DECAY)
# Decay the learning rate exponentially based on the number of steps. # Decay the learning rate exponentially based on the number of steps.
......
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