目录
一、 安装tensorboard二、 使用tensorboard1、首先导入模块2、初始化3、记录内容4、关闭三、可视化1、打开终端写命令行2、打开浏览器打开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内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!
X 关闭
X 关闭
- 15G资费不大降!三大运营商谁提供的5G网速最快?中国信通院给出答案
- 2联想拯救者Y70发布最新预告:售价2970元起 迄今最便宜的骁龙8+旗舰
- 3亚马逊开始大规模推广掌纹支付技术 顾客可使用“挥手付”结账
- 4现代和起亚上半年出口20万辆新能源汽车同比增长30.6%
- 5如何让居民5分钟使用到各种设施?沙特“线性城市”来了
- 6AMD实现连续8个季度的增长 季度营收首次突破60亿美元利润更是翻倍
- 7转转集团发布2022年二季度手机行情报告:二手市场“飘香”
- 8充电宝100Wh等于多少毫安?铁路旅客禁止、限制携带和托运物品目录
- 9好消息!京东与腾讯续签三年战略合作协议 加强技术创新与供应链服务
- 10名创优品拟通过香港IPO全球发售4100万股 全球发售所得款项有什么用处?