Update train.py
This commit is contained in:
1 parent
41ab1b23f1
commit
51a161d9e1
1 file changed
+1
-1
@@ -397,7 +397,7 @@ if __name__ == '__main__':
|
||||
# Train
|
||||
if not opt.evolve:
|
||||
print('Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/')
|
||||
tb_writer = SummaryWriter(log_dir=increment_dir('runs/exp', opt.name))
|
||||
tb_writer = SummaryWriter(log_dir=increment_dir('runs' + os.sep + 'exp', opt.name))
|
||||
if opt.hyp: # update hyps
|
||||
with open(opt.hyp) as f:
|
||||
hyp.update(yaml.load(f, Loader=yaml.FullLoader))
|
||||
|
||||
Reference in new issue
Block a user