mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

deepcopy map

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

4
codes/sweep.py
View File

@ -1,3 +1,4 @@
import copy
import functools
import os
from multiprocessing.pool import ThreadPool
@ -14,7 +15,6 @@ def launch_trainer(opt, opt_path=''):
trainer = Trainer()
opt['dist'] = False
trainer.rank = -1
torch.cuda.set_device(rank)
trainer.init(opt_path, opt, 'none')
trainer.do_training()
@ -37,7 +37,7 @@ if __name__ == '__main__':
opt = option.parse(base_opt, is_train=True)
all_opts = []
for i, (mod, mod_dict) in enumerate(modifications.items()):
nd = opt.copy()
nd = copy.deepcopy(opt)
nd.update(mod_dict)
opt['gpu_ids'] = [i]
nd['name'] = f'{nd["name"]}_{mod}'