與tensorflow模型與caffe模型不同,當前的pytorch沒有官方的直觀檢視網路結構的工具,google了下pytorch的網路解析的方法,發現可以將pytorch的model轉換成為events檔案使用tensorboard檢視,記錄之。
參考**
#-*-coding:utf-8-*-
import torch
import torchvision
from torch.autograd import variable
from tensorboardx import summarywriter
# 模擬輸入資料
input_data = variable(torch.rand(16, 3, 224, 224))
# 從torchvision中匯入已有模型
net = torchvision.models.resnet18()
# 宣告writer物件,儲存的資料夾
writer = summarywriter(log_dir='./log', comment='resnet18')
with writer:
writer.add_graph(net, (input_data,))
該**中14行宣告乙個writer物件,分別表示events存放的目錄,comment表示事件的title,然後使用如下的方式開啟tensorboard
tensorboard --logpath=d:\log --port=6006
然後按照命令列提示開啟即可。 使用Tensorboard工具檢視Loss損失率
在python程式中新增tensoboard一系列 我就不一一寫了。查詢資料就可以找到很多jexinr這方面的內容。今天我主要是想說下,如何在cmd中錄入命令,檢視。環境 win10 python開發環境 spyde程式設計客棧r tensorflow 這個是從 anaconda3 軟體中 的。準備...
py 列表的修改 檢視 排序
pv page view uv user view 定義乙個列表 service http samba nfs 通過索引,重新賦值 service 0 mysql 將索引值為0的字元修改為 mysql print service 通過切片 print service 2 顯示索引為2之前的元素 se...
遠端訪問Tensorboard
在學習tensorflow的過程中,視覺化工具tensorboard是必不可少的,但是由於本地主機計算效能等的侷限性,很多時候我們都是在遠端伺服器上執行tensor flow並訓練相關模型,所以學會相關遠端的操作至關重要,主要是ssh命令和scp命令,這裡我們只簡述一下如何訪問遠端的tensor b...