apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tensorboard comment=opt.name

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+1 -1
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)