Unverified Commit c1f8c0e2 authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

set default --required-batch-size-multiple to 1

parent e9d46fa0
......@@ -224,7 +224,7 @@ def add_dataset_args(parser, train=False, gen=False):
help='ignore too long or too short lines in valid and test set')
group.add_argument('--batch-size', '--max-sentences', type=int, metavar='N',
help='maximum number of sentences in a batch')
group.add_argument('--required-batch-size-multiple', default=8, type=int, metavar='N',
group.add_argument('--required-batch-size-multiple', default=1, type=int, metavar='N',
help='batch size will be a multiplier of this value')
group.add_argument('--data-buffer-size', default=10, type=int,
help='Number of batches to preload')
......
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