pytorch tensorboard可视化的使用详解
一、 安装tensorboard
直接pip即可:
pip install tensorboard
这里注意，使用pytorch，并不需要额外安装tensorflow。
二、 使用tensorboard
记录训练的loss和测试的accuracy：
1、首先导入模块
from torch.utils.tensorboard import SummaryWriter
2、初始化
writer = SummaryWriter('./log')记录的内容保存在文件夹log里面。
3、记录内容
writer.add_scalar('loss', loss, step)
writer.add_scalar('accuracy', accuracy, step)
这里step是指步数，loss为训练误差，accuracy为测试准确率，按自己需求记录即可。
4、关闭
writer.close()
在文件夹会log里会生成events.out.tfevents的文件，每次运行，如：
三、可视化
1、打开终端写命令行
在本地打开终端，进入虚拟环境，写以下命令行：
tensorboard --logdir=./log
这里需要注意一下，log为刚才保存文件的文件夹，可以是相对路径，也可以是绝对路径，只要稍微注意格式即可，以下有几个注意事项：
（1）、路径./log不要加双引号,，如：–logdir=“./log”
（2）、不要使用双等号，有些博客使用了双等号，其实是错误的，如：–logdir==./log
以上写法均是错误的，正确的写法有两种，一种就是上面例子使用单个等号，另外一种是使用空格，如：
tensorboard --logdir ./log
参数除了logdir外，常用的还有：host、port。
host：ip地址
port: 端口
tensorboard的默认是：
tensorboard --logdir=./log --host=127.0.0.1 --port=6006
有些博客说预防端口被占用，常常修改端口，其实一般不需要，host和port默认就好，只需写logdir就好，即按照开头那样写即可，回车后，有：
注意：这里需要注意一下，不要着急按下”Ctrl+C“，不然无法打开http://localhost:6006/，当不需要打开http://localhost:6006/，才按下”Ctrl+C“，即断开端口。
2、打开浏览器打开tensorboard可视化
当不需要可视化了，可以去终端按下”Ctrl+C“，关闭端口。
到此这篇关于pytorch tensorboard可视化的使用详解的文章就介绍到这了,更多相关pytorch tensorboard内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
python的scrapy框架之Pipeline文件的用法详解
这篇文章主要介绍了python的scrapy框架之Pipeline文件的用法详解,Pipeline是一个独立的模块,用于处理从Spider中提取的Item对象,实现对数据的进一步处理、存储和清洗等操作,下面将详细介绍Scrapy框架中Pipeline的用法,需要的朋友可以参考下2023-10-10
这篇文章主要介绍了Python seaborn barplot画图案例，文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下2022-07-07
这篇文章主要介绍了Python 并行加速技巧分享，文章围绕文章主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下2022-05-05
这篇文章主要介绍了分布式全文检索引擎ElasticSearch原理及使用实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-11-11
@dataclass 是 Python 3.7 引入的一个装饰器,用于方便地定义符合数据类协议的类,本文主要介绍了python3中@dataclass的实现示例,感兴趣的可以了解一下2024-02-02
python 使用elasticsearch 实现翻页的三种方式
这篇文章主要介绍了python 使用elasticsearch 实现翻页的三种方式，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-07-07
今天小编就为大家分享一篇python 类之间的参数传递方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
这篇文章主要介绍了Python星号*与**用法,结合实例形式较为详细的分析了Python中的星号*与**在函数参数及数值运算中的相关使用技巧,需要的朋友可以参考下2018-02-02
这篇文章主要介绍了Python检测QQ在线状态的方法,涉及Python通过第三方平台检测QQ在线状态的技巧,非常简单实用,需要的朋友可以参考下2015-05-05
今天小编就为大家分享一篇django manage.py扩展自定义命令方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
最新评论