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

Allow specification of wandb run name

Browse Source
This commit is contained in:
James Betker 2021-11-22 17:31:29 -07:00
parent 01e635168b
commit 3b5c3d85d8
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/train.py
View File

@ -85,7 +85,8 @@ class Trainer:
import wandb
os.makedirs(os.path.join(opt['path']['log'], 'wandb'), exist_ok=True)
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
seed = opt['train']['manual_seed']