tensorboard notice
This commit is contained in:
parent
05ae6e8499
commit
4fa3fd2df3
2
train.py
2
train.py
|
@ -416,7 +416,7 @@ if __name__ == '__main__':
|
|||
from torch.utils.tensorboard import SummaryWriter
|
||||
|
||||
tb_writer = SummaryWriter()
|
||||
print("Tensorboard found. Run 'tensorboard --logdir=runs' to view at http://localhost:6006/")
|
||||
print("Run 'tensorboard --logdir=runs' to view tensorboard at http://localhost:6006/")
|
||||
except:
|
||||
pass
|
||||
|
||||
|
|
Loading…
Reference in New Issue