deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use fork instead

Browse Source
This commit is contained in:
James Betker 2022-02-11 11:22:25 -07:00
parent 4abc094b47
commit 496fb81997
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
codes/sweep.py
View File

@ -48,5 +48,11 @@ if __name__ == '__main__':
nd['path'][k] = p.replace(base_path, f'{base_path}\\{mod}') nd['path'][k] = p.replace(base_path, f'{base_path}\\{mod}')
all_opts.append(nd) all_opts.append(nd)
with ThreadPool(len(modifications)) as pool: for i in range(1,len(modifications)):
list(pool.imap(functools.partial(launch_trainer, opt_path=base_opt), all_opts)) pid = os.fork()
if pid == 0:
rank = i
break
else:
rank = 0
launch_trainer(all_opts[i], base_opt)