tensorboard檢視pytorch模型

2022-04-28 23:12:18 字數 804 閱讀 5922

與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...