This commit is contained in:
Steve Nyemba 2021-04-13 17:43:43 -05:00
parent be55b14e2b
commit 567671c43e
1 changed files with 2 additions and 2 deletions

View File

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