一 寫(xiě)在前面
未經(jīng)允許，不得轉(zhuǎn)載岗屏，謝謝~~~
一個(gè)暑假都木有更新博客（/認(rèn)真捂臉.jpg）
馬上就開(kāi)學(xué)了，新學(xué)期加油哦~~~
這篇文章主要學(xué)習(xí)一下PyTorch如何用Tensorboard進(jìn)行模型漱办，數(shù)據(jù)以及訓(xùn)練過(guò)程的可視化这刷。
二 常用功能
2.1 TensorBoard基本設(shè)置
- 從
torch.utils中導(dǎo)入tensorboard中的SummaryWriter類;
from torch.utils.tensorboard import SummaryWriter
- 實(shí)例化用來(lái)記錄信息到TnesorBoard的
SummaryWriter類對(duì)象；
writer = SummaryWriter('logs/exp1/tensorboard')
- 其中
logs/exp1/tensorboard表示tensorboard的保存路徑娩井。
2.2 將內(nèi)容寫(xiě)入TensorBoard
1 寫(xiě)入圖像
writer.add_image('my_images', img)
- 其中
img為想要存入的圖像暇屋；
2 寫(xiě)入模型
writer.add_graph(net, images)
- 其中
net為網(wǎng)絡(luò)模型； - 其中
images為網(wǎng)絡(luò)模型的輸入撞牢；
3 寫(xiě)入網(wǎng)絡(luò)訓(xùn)練的loss等內(nèi)容
writer.add_scalar('loss', loss)
- 其中
loss為loss率碾； - 當(dāng)然也可以寫(xiě)入其他的標(biāo)量；
2.3 TensorBoard內(nèi)容查看
- 模型訓(xùn)練完成之后會(huì)在你設(shè)置的目標(biāo)文件夾下生成一個(gè)tensorboard文件屋彪。
$ ls logs/exp1/tensorboard/
events.out.tfevents.1566455334.bigvid-g01
- 找到這個(gè)文件所宰，然后運(yùn)行命令：
tensorboard --logdir=logs/exp1
- 然后根據(jù)命令行提示：
TensorBoard 1.8.0 at http://bigvid-g01:6006 (Press CTRL+C to quit)，把自己的服務(wù)器IP替換進(jìn)去貼到瀏覽器就可以看到了：
scalars
images
三 寫(xiě)在最后
這個(gè)用起來(lái)確實(shí)很方便畜挥，可以省去自己保存loss然后處理數(shù)據(jù)的煩惱仔粥，確實(shí)還不錯(cuò)~
這篇文章目前只嘗試了scalar和images的寫(xiě)入，其實(shí)還可以對(duì)模型以及數(shù)據(jù)的低維特征進(jìn)行可視化~
更多的細(xì)節(jié)還請(qǐng)參考官網(wǎng)-VISUALIZING MODELS, DATA, AND TRAINING WITH TENSORBOARD哦~~(>ω<*)
