From 51a161d9e1427ff0f094641760816133a358ef0d Mon Sep 17 00:00:00 2001
From: Laughing <61612323+Laughing-q@users.noreply.github.com>
Date: Fri, 10 Jul 2020 11:37:58 +0800
Subject: [PATCH] Update train.py
---
train.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train.py b/train.py
index bba0883..f67933b 100644
--- a/train.py
+++ b/train.py
@@ -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))