|
|
|
@ -298,11 +298,11 @@ if __name__ == '__main__' :
|
|
|
|
for id in ''.join(content) :
|
|
|
|
for id in ''.join(content) :
|
|
|
|
args['partition'] = id
|
|
|
|
args['partition'] = id
|
|
|
|
args['data'] = pd.DataFrame(DATA[(int(id))],columns=args['columns'])
|
|
|
|
args['data'] = pd.DataFrame(DATA[(int(id))],columns=args['columns'])
|
|
|
|
if args['num_gpu'] > 0 :
|
|
|
|
if args['num_gpu'] > 1 :
|
|
|
|
args['gpu'] = id
|
|
|
|
args['gpu'] = id
|
|
|
|
else:
|
|
|
|
else:
|
|
|
|
args['gpu']=0
|
|
|
|
args['gpu']=0
|
|
|
|
args['num_gpu']=1
|
|
|
|
|
|
|
|
job = Process(target=make,args=(args,))
|
|
|
|
job = Process(target=make,args=(args,))
|
|
|
|
job.name = 'generator # '+str(id)
|
|
|
|
job.name = 'generator # '+str(id)
|
|
|
|
job.start()
|
|
|
|
job.start()
|
|
|
|
|
