xx

5 years ago · 6f51eedca8
parent ff6ae5a622
commit 6f51eedca8
1 changed files with 2 additions and 2 deletions
--- 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()