|
|
|
@ -634,14 +634,14 @@ if __name__ == '__main__' :
|
|
|
|
#
|
|
|
|
#
|
|
|
|
if GPU_CHIPS and 'all-chips' in SYS_ARGS:
|
|
|
|
if GPU_CHIPS and 'all-chips' in SYS_ARGS:
|
|
|
|
index = 0
|
|
|
|
index = 0
|
|
|
|
|
|
|
|
print (['... launching ',len(GPU_CHIPS),' jobs',args['context']])
|
|
|
|
for _gpu in GPU_CHIPS :
|
|
|
|
for _gpu in GPU_CHIPS :
|
|
|
|
_args = copy.deepcopy(args)
|
|
|
|
_args = copy.deepcopy(args)
|
|
|
|
_args['gpu'] = [int(_gpu)]
|
|
|
|
_args['gpu'] = [int(_gpu)]
|
|
|
|
_args['partition'] = index
|
|
|
|
_args['partition'] = index
|
|
|
|
index += 1
|
|
|
|
index += 1
|
|
|
|
make = lambda _params: (Components()).train(**_params)
|
|
|
|
make = lambda _params: (Components()).train(**_params)
|
|
|
|
job = Process(target=make,args=( dict(_args),))
|
|
|
|
job = Process(target=make,args=( _args,))
|
|
|
|
job.name = 'Trainer # ' + str(index)
|
|
|
|
job.name = 'Trainer # ' + str(index)
|
|
|
|
job.start()
|
|
|
|
job.start()
|
|
|
|
jobs.append(job)
|
|
|
|
jobs.append(job)
|
|
|
|
|
