可视化工具

Tensorboard

• 导入tensorboard

from torch.utils.tensorboard import SummaryWriter

• 创建board，添加图表、图片、模型等

# 调用时在根目录创建文件夹log
board = SummaryWriter("log")

# 添加图表
board.add_scalar("tag1", y_label, x_label)

# 添加图片
# 注：若img_array不是CWH，则末尾追加dataformats
board.add_image("tag2", img_array, index)

# 添加模型
board.add_gragh(model, input)

board.close()

• 实例化（在终端启动Tensorboard，端口8080）

tensorboard --logdir=log --port=8080