Skip to content

Commit 46c0456

Browse files
authored
lower default batch producer queue size (#1716)
1 parent 0e6fbfd commit 46c0456

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

onmt/opts.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -375,7 +375,7 @@ def train_opts(parser):
375375
help="IP of master for torch.distributed training.")
376376
group.add('--master_port', '-master_port', default=10000, type=int,
377377
help="Port of master for torch.distributed training.")
378-
group.add('--queue_size', '-queue_size', default=400, type=int,
378+
group.add('--queue_size', '-queue_size', default=40, type=int,
379379
help="Size of queue for each process in producer/consumer")
380380

381381
group.add('--seed', '-seed', type=int, default=-1,

0 commit comments

Comments
 (0)