Unverified 提交 51a161d9 authored 作者: Laughing's avatar Laughing 提交者: GitHub

Update train.py

上级 41ab1b23
...@@ -397,7 +397,7 @@ if __name__ == '__main__': ...@@ -397,7 +397,7 @@ if __name__ == '__main__':
# Train # Train
if not opt.evolve: if not opt.evolve:
print('Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/') 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 if opt.hyp: # update hyps
with open(opt.hyp) as f: with open(opt.hyp) as f:
hyp.update(yaml.load(f, Loader=yaml.FullLoader)) hyp.update(yaml.load(f, Loader=yaml.FullLoader))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论