Skip to content

Commit

Permalink
Update default number of processes to 1 (#172)
Browse files Browse the repository at this point in the history
Co-authored-by: Marcos Pedro Ferreira Leal Silva <[email protected]>
  • Loading branch information
marcossilva and Marcos Pedro Ferreira Leal Silva authored Feb 24, 2022
1 parent ed73257 commit b2616d7
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -167,7 +167,7 @@ def main(args, path_to_candidate_bonds):
parser.add_argument('-rcb', '--reaction-center-batch-size', type=int, default=200,
help='Batch size to use for preparing candidate bonds from a trained '
'model on reaction center prediction')
parser.add_argument('-np', '--num-processes', type=int, default=8,
parser.add_argument('-np', '--num-processes', type=int, default=1,
help='Number of processes to use for data pre-processing')
parser.add_argument('-nw', '--num-workers', type=int, default=100,
help='Number of workers to use for data loading in PyTorch data loader')
Expand Down

0 comments on commit b2616d7

Please sign in to comment.