使用pytorch進行模型訓練時,僅僅通過觀察print的輸出是不夠的,有時候需要通過一些專業的工具來將損失值,訓練過程中產生的影象等資料展示出來,這個時候就可以借助visdom來實現資料的視覺化。
官方教程
示例**
pip install visdom
python -m visdom.server
如果執行完出現
downloading scripts,this may take a little while
等很久都沒有反應的話可以參考這裡,或者直接fq。
通過瀏覽器檢視視覺化資料:http://localhost:8097/
通過如下**可以畫一條曲線和一張
from visdom import效果圖visdom
import
torch
vis = visdom(env='
main')
x = torch.linspace(0, 10, 1000)
y = x + 5vis.line(x=x, y=y)
vis.image(torch.ones((3, 100, 100)) * 127)
損失值等資料肯定是動態展示的,所以需要在每次出新的損失值時,動態的新增到指定視窗中去
from visdom import效果圖visdom
import
torch
class
logger():
def__init__
(self):
self.vis =visdom()
self.cur_iter =0
deflog(self, loss):
self.vis.line(x=[self.cur_iter], y=[loss], win='
loss
', update='')
self.cur_iter += 1logger =logger()
for i in range(50000):
loss = torch.rand([1])
logger.log(loss)
效果圖
如果後續還需要定製自己用來展示資料的方式,可以參考官方教程,在原來的基礎上,新增新的**。
PyTorch視覺化工具visdom遠端方法
visdom是由facebook開發的pytorch訓練過程視覺化工具,類似於tensorboard。本文主要記錄在遠端伺服器上跑pytorch模型時如何在本地訪問visdom的結果。ssh遠端伺服器時繫結本地埠與遠端伺服器端口 ssh l localhost p username remote h...
詳解PyTorch視覺化工具visdom(一)
在深度學習領域,模型訓練是乙個必須的過程,因此常常需要實時監聽並視覺化一些資料,如損失值loss,正確率acc等。在tensorflow中,最常使用的工具非tensorboard莫屬 在pytorch中,也有類似的tensorboardx,但據說其在張量資料載入的效率方面不如visdom。visdo...
Pytorch視覺化工具Visdom超簡單入門
visdom的github repo 知乎乙個教程 上次寫過visdom在伺服器端啟動的博文 這次簡單介紹下在python 裡用visdom 非常簡單 import visdom vis visdom.visdom env visual eye port 10041 env後是環境名稱,port是伺...