網站首頁 編程語言 正文
一、 安裝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“,關閉端口。
原文鏈接:https://blog.csdn.net/Caesar6666/article/details/127245418
相關推薦
- 2022-02-11 為了兼容IE,配置Babel+Webpack
- 2022-12-30 React高階組件使用教程詳解_React
- 2022-07-04 C語言植物大戰數據結構堆排序圖文示例_C 語言
- 2022-05-29 解決Docker容器下不能使用vim命令的問題_docker
- 2023-05-23 Numpy中扁平化函數ravel()和flatten()的區別詳解_python
- 2022-07-16 SpringMVC 傳遞參數
- 2022-02-16 C語言實現wave波形_C 語言
- 2022-10-14 el-tree 懶加載,默認加載N級.異步加載子節點
- 最近更新
-
- window11 系統安裝 yarn
- 超詳細win安裝深度學習環境2025年最新版(
- Linux 中運行的top命令 怎么退出?
- MySQL 中decimal 的用法? 存儲小
- get 、set 、toString 方法的使
- @Resource和 @Autowired注解
- Java基礎操作-- 運算符,流程控制 Flo
- 1. Int 和Integer 的區別,Jav
- spring @retryable不生效的一種
- Spring Security之認證信息的處理
- Spring Security之認證過濾器
- Spring Security概述快速入門
- Spring Security之配置體系
- 【SpringBoot】SpringCache
- Spring Security之基于方法配置權
- redisson分布式鎖中waittime的設
- maven:解決release錯誤:Artif
- restTemplate使用總結
- Spring Security之安全異常處理
- MybatisPlus優雅實現加密?
- Spring ioc容器與Bean的生命周期。
- 【探索SpringCloud】服務發現-Nac
- Spring Security之基于HttpR
- Redis 底層數據結構-簡單動態字符串(SD
- arthas操作spring被代理目標對象命令
- Spring中的單例模式應用詳解
- 聊聊消息隊列,發送消息的4種方式
- bootspring第三方資源配置管理
- GIT同步修改后的遠程分支