Pytorch如何打印與Keras的model.summary()類似的輸出(最新推薦)
更新時間：2023年07月22日 12:00:16 作者：Xminyang
這篇文章主要介紹了Pytorch如何打印與Keras的model.summary()類似的輸出,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下
1 Keras的model.summary()
2 Pytorch實現(xiàn)
2.1 安裝torchsummary包
pip install torchsummary
2.2 代碼
import torch
import torch.nn as nn
import torch.nn.functional as F
from torchsummary import summary
class Net(nn.Module):
def __init__(self):
super(Net, self).__init__()
self.conv1 = nn.Conv2d(1, 10, kernel_size=5)
self.conv2 = nn.Conv2d(10, 20, kernel_size=5)
self.conv2_drop = nn.Dropout2d()
self.fc1 = nn.Linear(320, 50)
self.fc2 = nn.Linear(50, 10)
def forward(self, x):
x = F.relu(F.max_pool2d(self.conv1(x), 2))
x = F.relu(F.max_pool2d(self.conv2_drop(self.conv2(x)), 2))
x = x.view(-1, 320)
x = F.relu(self.fc1(x))
x = F.dropout(x, training=self.training)
x = self.fc2(x)
return F.log_softmax(x, dim=1)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") # PyTorch v0.4.0
model = Net().to(device)
summary(model, (1, 28, 28))2.3 運行結(jié)果
參考材料：
Keras style model.summary() in PyTorch
How do I print the model summary in Pytorch
到此這篇關(guān)于Pytorch如何打印與Keras的model.summary()類似的輸出的文章就介紹到這了,更多相關(guān)Pytorch打印內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Hugging Face Transformers 是自然語言處理（NLP）領(lǐng)域最流行的開源庫之一,支持基于 Transformer 架構(gòu)的預(yù)訓(xùn)練模型（如 BERT、GPT、T5 等）,覆蓋文本、圖像、音頻等多模態(tài)任務(wù),本文介紹Python 中的 Transformer 庫及使用方法,感興趣的朋友一起看看吧2025-04-04
Python多線程采集二手房源數(shù)據(jù)信息流程詳解
這篇文章主要介紹了Python多線程采集二手房源數(shù)據(jù)信息流程，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)吧2023-05-05
這篇文章主要介紹了Python3.6日志Logging模塊簡單用法,結(jié)合實例形式分析了Python3.6環(huán)境下日志Logging模塊設(shè)置格式、文件流輸出相關(guān)操作技巧,需要的朋友可以參考下2018-06-06
詳解appium自動化測試工具(monitor、uiautomatorviewer)
這篇文章主要介紹了詳解appium自動化測試工具（monitor、uiautomatorviewer），文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-01-01
這篇文章主要為大家詳細(xì)介紹了如何使用python寫個喚醒睡眠電腦的腳本,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2025-01-01
本文介紹了在Ubuntu系統(tǒng)上卸載和安裝PyCharm的具體步驟,包括刪除安裝包、卸載文件夾、刪除配置文件夾、從官網(wǎng)下載并安裝Professional版、設(shè)置快捷方式以及修改啟動器等內(nèi)容2026-05-05
這篇文章主要為大家詳細(xì)介紹了python實現(xiàn)停車管理系統(tǒng)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-11-11
這篇文章主要為大家詳細(xì)介紹了使用tensorflow實現(xiàn)線性回歸，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-09-09
什么是好用的代碼呢？其實就是代碼質(zhì)量比較高，如何評價代碼質(zhì)量的高低呢？最常用的、最重要的評價標(biāo)準(zhǔn)，就是代碼的可維護(hù)性、可讀性、可擴(kuò)展性、靈活性、簡潔性、可復(fù)用性、可測試性2021-09-09
這篇文章主要給大家介紹了關(guān)于Python中Yield的基本用法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-10-10