Merge pull request #37 from ShadiRahimian:patch-1

PiperOrigin-RevId: 240601653
This commit is contained in:
A. Unique TensorFlower 2019-03-27 11:13:30 -07:00
commit a8e54d928c

View file

@ -80,8 +80,8 @@ def main(argv):
if q > 1: if q > 1:
raise app.UsageError('N must be larger than the batch size.') raise app.UsageError('N must be larger than the batch size.')
orders = ([1.25, 1.5, 1.75, 2., 2.25, 2.5, 3., 3.5, 4., 4.5] + range(5, 64) + orders = ([1.25, 1.5, 1.75, 2., 2.25, 2.5, 3., 3.5, 4., 4.5] +
[128, 256, 512]) list(range(5, 64)) + [128, 256, 512])
steps = int(math.ceil(FLAGS.epochs * FLAGS.N / FLAGS.batch_size)) steps = int(math.ceil(FLAGS.epochs * FLAGS.N / FLAGS.batch_size))