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
|
from torch.utils.tensorboard import SummaryWriter
|
||||||
|
|
||||||
tb_writer = 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:
|
except:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue