From 0410435a8a6820bc4773ec01e2e2328374d6044e Mon Sep 17 00:00:00 2001 From: Yin Li Date: Thu, 25 Mar 2021 17:05:36 -0400 Subject: [PATCH] Change default number of dataloader workers as suggested by pytorch --- map2map/args.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/map2map/args.py b/map2map/args.py index c80c5e0..ef0f146 100644 --- a/map2map/args.py +++ b/map2map/args.py @@ -81,10 +81,9 @@ def add_common_args(parser): # "batches" is kept for now for backward compatibility parser.add_argument('--batch-size', '--batches', type=int, required=True, help='mini-batch size, per GPU in training or in total in testing') - parser.add_argument('--loader-workers', default=-8, type=int, + parser.add_argument('--loader-workers', default=8, type=int, help='number of subprocesses per data loader. ' - '0 to disable multiprocessing; ' - 'negative number to multiply by the batch size') + '0 to disable multiprocessing') parser.add_argument('--callback-at', type=lambda s: os.path.abspath(s), help='directory of custorm code defining callbacks for models, ' @@ -186,8 +185,7 @@ def int_tuple(s): def set_common_args(args): - if args.loader_workers < 0: - args.loader_workers *= - args.batch_size + pass def set_train_args(args):