From f0ac353be8db290283d1dc385e5985ed0ff2de26 Mon Sep 17 00:00:00 2001 From: Marcos Pedro Ferreira Leal Silva Date: Wed, 23 Feb 2022 18:31:58 -0300 Subject: [PATCH] Update default number of processes to 1 --- .../rexgen_direct/candidate_ranking_train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/reaction_prediction/rexgen_direct/candidate_ranking_train.py b/examples/reaction_prediction/rexgen_direct/candidate_ranking_train.py index 213621c4..4832f0b7 100644 --- a/examples/reaction_prediction/rexgen_direct/candidate_ranking_train.py +++ b/examples/reaction_prediction/rexgen_direct/candidate_ranking_train.py @@ -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')