Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

tensorboard comment=opt.name

This commit is contained in:
Glenn Jocher 2020-04-21 12:19:29 -07:00
parent 4c4f4f4dd4
commit a5bd0fe567
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -420,7 +420,7 @@ if __name__ == '__main__':
tb_writer = None tb_writer = None
if not opt.evolve: # Train normally if not opt.evolve: # Train normally
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() tb_writer = SummaryWriter(comment=opt.name)
train() # train normally train() # train normally
else: # Evolve hyperparameters (optional) else: # Evolve hyperparameters (optional)