Pytorch如何打印与Keras的model.summary()类似的输出(最新推荐)

更新时间：2023年07月22日 12:00:16 作者：Xminyang

这篇文章主要介绍了Pytorch如何打印与Keras的model.summary()类似的输出,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下

1 Keras的model.summary()

在这里插入图片描述

2 Pytorch实现

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 运行结果

在这里插入图片描述

参考材料：

Keras style model.summary() in PyTorch

How do I print the model summary in Pytorch

到此这篇关于Pytorch如何打印与Keras的model.summary()类似的输出的文章就介绍到这了,更多相关Pytorch打印内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

Pycharm运行时总是跳出Python Console问题
这篇文章主要介绍了Pycharm运行时总是跳出Python Console问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2023-04-04
python 利用Pyinstaller打包Web项目
这篇文章主要介绍了python 利用Pyinstaller打包Web项目，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下
2020-10-10
pygame实现方块动画实例讲解
在本篇文章里小编给大家整理的是一篇关于pygame实现方块动画实例讲解内容，以后需要的朋友们可以学习参考下。
2021-12-12
详解PyTorch nn.Embedding() 嵌入
在自然语言处理（NLP）中,将文本序列转化为数字序列（tokenid）后,为了使模型能更好地理解这些数字背后的含义,引入了嵌入层（Embedding）通过简单的示例,可以看出Embedding的获取过程及其在理解语言中的关键作用
2024-11-11
Python真题案例之二分法查找详解
这篇文章主要介绍了python实操案例练习，本文给大家分享的案例中主要讲解了二分法查找，需要的小伙伴可以参考一下
2022-03-03
Python日志模块logging简介
这篇文章主要介绍了Python日志模块logging简介,本文讲解了Logger、Handler、Formatter、日志配置管理、通过文件配置管理日志等内容,需要的朋友可以参考下
2015-04-04
python常见模块与用法
这篇文章主要介绍了python常见模块与用法,文中有非常详细的代码示例,对正在学习python的小伙伴们有一定的帮助,需要的朋友可以参考下
2021-04-04
anaconda虚拟环境python sklearn库的安装过程
Anaconda是专注于数据分析的Python发行版本,包含了conda、Python等190多个科学包及其依赖项,这篇文章主要给大家介绍了关于anaconda虚拟环境python sklearn库的安装过程,需要的朋友可以参考下
2023-11-11
Django使用HttpResponse返回图片并显示的方法
今天小编就为大家分享一篇Django使用HttpResponse返回图片并显示的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-05-05
python GUI库图形界面开发之PyQt5信号与槽基本操作
这篇文章主要介绍了python GUI库图形界面开发之PyQt5信号与槽基本操作,需要的朋友可以参考下
2020-02-02