視覺化工具 Visdom

2022-09-20 00:51:13 字數 2410 閱讀 5929

使用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是伺...