Pytorch如何打印與Keras的model.summary()類似的輸出(最新推薦)
更新時間：2023年07月22日 12:00:16 作者：Xminyang
這篇文章主要介紹了Pytorch如何打印與Keras的model.summary()類似的輸出,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下
1 Keras的model.summary()
2 Pytorch實(shí)現(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 運(yùn)行結(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)文章
Pycharm運(yùn)行時總是跳出Python?Console問題
這篇文章主要介紹了Pycharm運(yùn)行時總是跳出Python?Console問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-04-04
python 利用Pyinstaller打包Web項(xiàng)目
這篇文章主要介紹了python 利用Pyinstaller打包Web項(xiàng)目，幫助大家更好的理解和使用python，感興趣的朋友可以了解下2020-10-10
pygame實(shí)現(xiàn)方塊動畫實(shí)例講解
在本篇文章里小編給大家整理的是一篇關(guān)于pygame實(shí)現(xiàn)方塊動畫實(shí)例講解內(nèi)容，以后需要的朋友們可以學(xué)習(xí)參考下。2021-12-12
在自然語言處理（NLP）中,將文本序列轉(zhuǎn)化為數(shù)字序列（tokenid）后,為了使模型能更好地理解這些數(shù)字背后的含義,引入了嵌入層（Embedding）通過簡單的示例,可以看出Embedding的獲取過程及其在理解語言中的關(guān)鍵作用2024-11-11
這篇文章主要介紹了python實(shí)操案例練習(xí)，本文給大家分享的案例中主要講解了二分法查找，需要的小伙伴可以參考一下2022-03-03
這篇文章主要介紹了Python日志模塊logging簡介,本文講解了Logger、Handler、Formatter、日志配置管理、通過文件配置管理日志等內(nèi)容,需要的朋友可以參考下2015-04-04
這篇文章主要介紹了python常見模塊與用法,文中有非常詳細(xì)的代碼示例,對正在學(xué)習(xí)python的小伙伴們有一定的幫助,需要的朋友可以參考下2021-04-04
anaconda虛擬環(huán)境python?sklearn庫的安裝過程
Anaconda是專注于數(shù)據(jù)分析的Python發(fā)行版本,包含了conda、Python等190多個科學(xué)包及其依賴項(xiàng),這篇文章主要給大家介紹了關(guān)于anaconda虛擬環(huán)境python?sklearn庫的安裝過程,需要的朋友可以參考下2023-11-11
Django使用HttpResponse返回圖片并顯示的方法
今天小編就為大家分享一篇Django使用HttpResponse返回圖片并顯示的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05
python GUI庫圖形界面開發(fā)之PyQt5信號與槽基本操作
這篇文章主要介紹了python GUI庫圖形界面開發(fā)之PyQt5信號與槽基本操作,需要的朋友可以參考下2020-02-02