Merge pull request #344 from Laughing-q/master
Browse files
train.py
CHANGED
@@ -397,7 +397,7 @@ if __name__ == '__main__':
|
|
397 |
# Train
|
398 |
if not opt.evolve:
|
399 |
print('Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/')
|
400 |
-
tb_writer = SummaryWriter(log_dir=increment_dir('runs
|
401 |
if opt.hyp: # update hyps
|
402 |
with open(opt.hyp) as f:
|
403 |
hyp.update(yaml.load(f, Loader=yaml.FullLoader))
|
|
|
397 |
# Train
|
398 |
if not opt.evolve:
|
399 |
print('Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/')
|
400 |
+
tb_writer = SummaryWriter(log_dir=increment_dir('runs' + os.sep + 'exp', opt.name))
|
401 |
if opt.hyp: # update hyps
|
402 |
with open(opt.hyp) as f:
|
403 |
hyp.update(yaml.load(f, Loader=yaml.FullLoader))
|