Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Change default number of loader workers

Browse Source
This commit is contained in:
Yin Li 2020-07-15 22:03:14 -04:00
parent 7d3a598080
commit ff4c2e12fd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
map2map/args.py
View File

@ -70,7 +70,7 @@ def add_common_args(parser):
parser.add_argument('--batches', type=int, required=True, parser.add_argument('--batches', type=int, required=True,
help='mini-batch size, per GPU in training or in total in testing') 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=-2, type=int,
help='number of subprocesses per data loader. ' help='number of subprocesses per data loader. '
'0 to disable multiprocessing; ' '0 to disable multiprocessing; '
'negative number to multiply by the batch size') 'negative number to multiply by the batch size')