From 6f51eedca80dde48264647a8fefb03d742bd2506 Mon Sep 17 00:00:00 2001 From: Steve Nyemba Date: Fri, 6 Mar 2020 14:45:22 -0600 Subject: [PATCH] xx --- pipeline.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pipeline.py b/pipeline.py index dfa0331..2ce90a9 100644 --- a/pipeline.py +++ b/pipeline.py @@ -298,11 +298,11 @@ if __name__ == '__main__' : for id in ''.join(content) : args['partition'] = id args['data'] = pd.DataFrame(DATA[(int(id))],columns=args['columns']) - if args['num_gpu'] > 0 : + if args['num_gpu'] > 1 : args['gpu'] = id else: args['gpu']=0 - args['num_gpu']=1 + job = Process(target=make,args=(args,)) job.name = 'generator # '+str(id) job.start()