1. 使用方法
- 写好代码,生成文件夹logs
- 到logs所在目录(当前目录)下,打开终端运行
tensorboard --logdir=logs
- 根据提示打开页面
http://localhost:6006/
2. 看图片 SummaryWriter().add_image()
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
writer = SummaryWriter('logs')
img_tensor = torch.randn((3, 3200, 1800))
writer.add_image('test', img_tensor)
writer.close()
2. 看模型 SummaryWriter().add_graph()
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter('logs')
writer.add_graph(MyModule(), torch.randn(2, 3, 256, 256))
writer.close()