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

sweep fix

Browse Source
This commit is contained in:
James Betker 2022-02-11 11:17:08 -07:00
parent 102142d1eb
commit 006add64c5
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
codes/sweep.py
View File

@ -9,8 +9,8 @@ from utils import options as option
def launch_trainer(opt, opt_path=''): def launch_trainer(opt, opt_path=''):
rank = opt['gpu_ids'][0] rank = opt['gpu_ids'][0]
os.environ['CUDA_VISIBLE_DEVICES'] = [rank] os.environ['CUDA_VISIBLE_DEVICES'] = str(rank)
print('export CUDA_VISIBLE_DEVICES=' + rank) print('export CUDA_VISIBLE_DEVICES=' + str(rank))
trainer = Trainer() trainer = Trainer()
opt['dist'] = False opt['dist'] = False
trainer.rank = -1 trainer.rank = -1
@ -23,7 +23,7 @@ if __name__ == '__main__':
Ad-hoc script (hard coded; no command-line parameters) that spawns multiple separate trainers from a single options Ad-hoc script (hard coded; no command-line parameters) that spawns multiple separate trainers from a single options
file, with a hard-coded set of modifications. file, with a hard-coded set of modifications.
""" """
base_opt = '../options/train_diffusion_tts.yml' base_opt = '../experiments/train_diffusion_tts6.yml'
modifications = { modifications = {
'baseline': {}, 'baseline': {},
'only_conv': {'networks': {'generator': {'kwargs': {'cond_transformer_depth': 4, 'mid_transformer_depth': 1}}}}, 'only_conv': {'networks': {'generator': {'kwargs': {'cond_transformer_depth': 4, 'mid_transformer_depth': 1}}}},
@ -49,4 +49,4 @@ if __name__ == '__main__':
all_opts.append(nd) all_opts.append(nd)
with ThreadPool(len(modifications)) as pool: with ThreadPool(len(modifications)) as pool:
list(pool.imap(functools.partial(launch_trainer, opt_path=base_opt), all_opts)) list(pool.imap(functools.partial(launch_trainer, opt_path=base_opt), all_opts))