Skip to content

Commit 6d50131

Browse files
committed
add multiprocessing option
1 parent fc970b6 commit 6d50131

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

GenNet_utils/Train_network.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -51,8 +51,8 @@ def train_classification(args):
5151
if args.workers > 1:
5252
multiprocessing = True
5353
else:
54-
pass
55-
54+
multiprocessing = False
55+
5656
check_data(datapath=datapath, genotype_path=genotype_path, mode=problem_type)
5757

5858
global weight_positive_class, weight_negative_class
@@ -238,7 +238,7 @@ def train_regression(args):
238238
if args.workers > 1:
239239
multiprocessing = True
240240
else:
241-
pass
241+
multiprocessing = False
242242

243243
check_data(datapath=datapath, genotype_path=genotype_path, mode=problem_type)
244244

0 commit comments

Comments
 (0)