如果想用pytorch來做神經(jīng)網(wǎng)絡(luò)學(xué)習(xí)和訓(xùn)練，那么不管想不想，你都得學(xué)visdom這個(gè)可視化工具，因?yàn)閯e人全部用這個(gè)玩意在進(jìn)行可視化顯示，如果不會(huì)用有時(shí)甚至別人的代碼都跑不起來。這里這么說似乎顯得有點(diǎn)怨念，這是真的，因?yàn)楸救艘婚_始并不想要去搞它。但后來學(xué)著學(xué)著發(fā)現(xiàn)這真是個(gè)神器，難怪那么多人一上來直接要求先裝上這個(gè)。

要用這個(gè)先要安裝，對(duì)于python模塊而言，安裝都是蠻簡(jiǎn)單的：

pip install visdom

安裝完每次要用直接輸入代碼打開：

python -m visdom.server

然后根據(jù)提示在瀏覽器中輸入相應(yīng)地址即可，默認(rèn)地址為：

http://localhost:8097/

使用示例如下：

import visdom
import torch
# 新建一個(gè)連接客戶端
# 指定env = 'test1'，默認(rèn)是'main',注意在瀏覽器界面做環(huán)境的切換
vis = visdom.Visdom(env='test1')
# 繪制正弦函數(shù)
x = torch.arange(1, 100, 0.01)
y = torch.sin(x)
vis.line(X=x,Y=y, win='sinx',opts={'title':'y=sin(x)'})
# 繪制36張圖片隨機(jī)的彩色圖片
vis.images(torch.randn(36,3,64,64).numpy(),nrow=6, win='imgs',opts={'title':'imgs'})

如果要用來繪制loss函數(shù)的變化趨勢(shì)，可以使用line函數(shù)，具體如下：

#繪制loss變化趨勢(shì)，參數(shù)一為Y軸的值，參數(shù)二為X軸的值，參數(shù)三為窗體名稱，參數(shù)四為表格名稱，參數(shù)五為更新選項(xiàng)，從第二個(gè)點(diǎn)開始可以更新
vis.line(Y=np.array([totalloss.item()]), X=np.array([traintime]),
                win=('train_loss'),
                opts=dict(title='train_loss'),
                update=None if traintime == 0 else 'append'
                )

總結(jié)

以上是生活随笔為你收集整理的visdom简易教程的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。