glenn-jocher commited on
Commit
17b9d7c
·
unverified ·
2 Parent(s): 59beae9 51a161d

Merge pull request #344 from Laughing-q/master

Browse files
Files changed (1) hide show
  1. train.py +1 -1
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/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))
 
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))