Skip to content

Commit f40854b

Browse files
Allow custom --evolve project names (#6567)
* Update train.py As see in #6463, modification on train in evolve process to allow custom save directory. * fix val * PEP8 whitespace around operator * Cleanup Co-authored-by: Glenn Jocher <[email protected]>
1 parent f627bc5 commit f40854b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

train.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -515,7 +515,8 @@ def main(opt, callbacks=Callbacks()):
515515
check_file(opt.data), check_yaml(opt.cfg), check_yaml(opt.hyp), str(opt.weights), str(opt.project) # checks
516516
assert len(opt.cfg) or len(opt.weights), 'either --cfg or --weights must be specified'
517517
if opt.evolve:
518-
opt.project = str(ROOT / 'runs/evolve')
518+
if opt.project == str(ROOT / 'runs/train'): # if default project name, rename to runs/evolve
519+
opt.project = str(ROOT / 'runs/evolve')
519520
opt.exist_ok, opt.resume = opt.resume, False # pass resume to exist_ok and disable resume
520521
opt.save_dir = str(increment_path(Path(opt.project) / opt.name, exist_ok=opt.exist_ok))
521522

0 commit comments

Comments
 (0)