tensorboard可视化工具汇总
- 导入依赖项
from torch.utils.tensorboard import SummaryWriter
writer=SummaryWriter()
....
# 使用完毕关闭
writer.close()
- 可视化计算图
writer.add_grapg(model,input)
- 可视化单个标量
writer.add_scalar("train_loss",loss,epoch)
- 在一个图中同时画出两个两个变量的曲线对比
writer.add_scalars("Loss",{"Training loss":total_loss*1.0/total},e)
writer.add_scalars("Loss",{"Test loss":all_loss*1.0/total},e)