pytorch, tensorboard,

pytorch使用 tensorboard 教學

Champion Champion Follow Jun 07, 2021 · 1 min read
pytorch使用 tensorboard 教學
Share this

以前訓練模型都是硬train一發,看看模型效果好不好,但是我們如果能透過視覺化模型的權重以及loss的趨勢來判斷 模型訓練得好不好,就不會像盲人摸象一樣,無法判斷模型哪裡有問題,所以今天要來介紹tensorboard的使用~~~

安裝tensorboard

安裝tensorboard的版本需要在1.15版本以上才可以在pytorch使用

啟動tensorboard

logdir為tensorboard輸出數據的地方

使用教學

導入tensorboard以及其他分析套件

實例化SummaryWriter,這個的作用就是將等等要寫入的數據用特定的格式寫進指定的資料夾,

資料夾路徑預設為./run,下面的add_scalar函數在訓練的時候會常常用到,可以將訓練中的loss以及accuracy寫入tensorboard,後續再作分析。

啟動tensorboard後就可以在SCALARS的選項中看到我們剛剛在訓練過程中的準確度和loss,路徑前面使用一樣的名字就會被歸在同一類


(圖一)

分布分析

使用add_histogram可以用來做資料分布的視覺化。


(圖二)

文字分析

使用add_text這個函式就可以輸出文字到tensorboard,使用方式和add_scalar一樣,前面參數是分類的tag,後面是要輸出的文字。


(圖三)

圖片分析

常常我們會對資料去做一些視覺化,同樣也可以透過tensorboard的add_figure函數來實現,一樣加上分類tag,然後使用matplotlib畫圖然後輸出。


(圖四)

那以上就是關於如何使用tensorboard的介紹,如果有什麼問題都可以在下面進行留言!!!如果喜歡我的文章可以幫我拍拍手哦~~~~

Join Newsletter
Get the latest news right in your inbox. We never spam!
Champion
Written by Champion Follow
Hi, I am Champion, the author of CHAMPION.AI, I will share some AI and investment articles. I hope you like it.