【Pytorch】torch.utils.tensorboard

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()