"tests/vscode:/vscode.git/clone" did not exist on "b4b9af6b1f1fb72e5fafd55c737b23bed31f7b3a"
Unverified Commit c8032908 authored by cheerss's avatar cheerss Committed by GitHub
Browse files

update the calculation of num_batches_per_epoch

the number of batches per epoch also depends on the number of gpus. it should be `num_batches_per_epoch = (cifar10.NUM_EXAMPLES_PER_EPOCH_FOR_TRAIN / FLAGS.batch_size / FLAGS.num_gpus)`
parent 82e783e3
......@@ -147,7 +147,7 @@ def train():
# Calculate the learning rate schedule.
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 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