TensorBoard的一些认识


文章目录


一、TensorBoard简介

机器学习可视化工具,用来记录机器学习过程。
TensorBoard界面展示:
在这里插入图片描述

二、TensorBoard安装与运行

#安装命令

pip install tensorboard

安装后再pycharm终端输入

tensorboard --help

若能正常显示,则说明运行成功。

三、Pytorch使用TensorBoard

Pytorch使用Tensorboard主要用到了三个API:

. SummaryWriter:这个用来创建一个log文件,TensorBoard面板查看时,也是需要选择查看那个log文件。
. add_something: 向log文件里面增添数据。例如可以通过add_scalar增添折线图数据,add_image可以增添图片。
. close:当训练结束后,我们可以通过close方法结束log写入。

四、总结

总的来说,pytorch使用TensorBoard的过程就是:使用SummaryWriter()创建一个log文件,然后使用add_something()或者add_image()往log文件里面添加数据,之后在tensorboard中看到的数据来源于这个log文件。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值