tensorboardX可视化
tensorboardX是基于tensorboard所做的一个用于pytorch数据可视化的一款工具。可以支持标量、图像、音频、文本、pytorch中搭建的网络结构等等。
安装：
pip install tensorboardX绘制标量简单实例：
from tensorboardX import SummaryWriter
writer = SummaryWriter("log")
for i in range(100):
writer.add_scalar("a", i, global_step=i)
writer.add_scalar("b", i**2, global_step=i)
writer.close()这时候会在当前目录生成一个log文件夹，在terminal命令行输入：
tensorboard --logdir log/会返回：
打开其中的链接：
可以看到这是我们创建的a和b的标量。
本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
腾讯云开发者
