update log_dir to runs/exp #107

train.py

@@ -396,7 +396,7 @@ if __name__ == '__main__':
     # Train
     if not opt.evolve:
         print('Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/')
-        tb_writer = SummaryWriter(comment=opt.name)
+        tb_writer = SummaryWriter(log_dir=increment_dir('runs/exp', opt.name))
         if opt.hyp:  # update hyps
             with open(opt.hyp) as f:
                 hyp.update(yaml.load(f, Loader=yaml.FullLoader))