Steve L. Nyemba 5 years ago
parent ff6ae5a622
commit 6f51eedca8

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

Loading…
Cancel
Save