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()