forked from mrq/DL-Art-School
Allow specification of wandb run name
This commit is contained in:
parent
01e635168b
commit
3b5c3d85d8
|
@ -85,7 +85,8 @@ class Trainer:
|
||||||
import wandb
|
import wandb
|
||||||
os.makedirs(os.path.join(opt['path']['log'], 'wandb'), exist_ok=True)
|
os.makedirs(os.path.join(opt['path']['log'], 'wandb'), exist_ok=True)
|
||||||
project_name = opt_get(opt, ['wandb_project_name'], opt['name'])
|
project_name = opt_get(opt, ['wandb_project_name'], opt['name'])
|
||||||
wandb.init(project=project_name, dir=opt['path']['log'], config=opt)
|
run_name = opt_get(opt, ['wandb_run_name'], None)
|
||||||
|
wandb.init(project=project_name, dir=opt['path']['log'], config=opt, name=run_name)
|
||||||
|
|
||||||
#### random seed
|
#### random seed
|
||||||
seed = opt['train']['manual_seed']
|
seed = opt['train']['manual_seed']
|
||||||
|
|
Loading…
Reference in New Issue
Block a user