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打印内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

Python Transformer 库安装配置及使用方法
Hugging Face Transformers 是自然语言处理（NLP）领域最流行的开源库之一,支持基于 Transformer 架构的预训练模型（如 BERT、GPT、T5 等）,覆盖文本、图像、音频等多模态任务,本文介绍Python 中的 Transformer 库及使用方法,感兴趣的朋友一起看看吧
2025-04-04
Python多线程采集二手房源数据信息流程详解
这篇文章主要介绍了Python多线程采集二手房源数据信息流程，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习吧
2023-05-05
Python3.6日志Logging模块简单用法示例
这篇文章主要介绍了Python3.6日志Logging模块简单用法,结合实例形式分析了Python3.6环境下日志Logging模块设置格式、文件流输出相关操作技巧,需要的朋友可以参考下
2018-06-06
详解appium自动化测试工具(monitor、uiautomatorviewer)
这篇文章主要介绍了详解appium自动化测试工具（monitor、uiautomatorviewer），文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-01-01
python写个唤醒睡眠电脑的脚本
这篇文章主要为大家详细介绍了如何使用python写个唤醒睡眠电脑的脚本,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下
2025-01-01
ubuntu中pycharm卸载与安装教程
本文介绍了在Ubuntu系统上卸载和安装PyCharm的具体步骤,包括删除安装包、卸载文件夹、删除配置文件夹、从官网下载并安装Professional版、设置快捷方式以及修改启动器等内容
2026-05-05
python实现停车管理系统
这篇文章主要为大家详细介绍了python实现停车管理系统，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-11-11
使用tensorflow实现线性回归
这篇文章主要为大家详细介绍了使用tensorflow实现线性回归，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-09-09
Python编程应用设计原则详解
什么是好用的代码呢？其实就是代码质量比较高，如何评价代码质量的高低呢？最常用的、最重要的评价标准，就是代码的可维护性、可读性、可扩展性、灵活性、简洁性、可复用性、可测试性
2021-09-09
Python中Yield的基本用法
这篇文章主要给大家介绍了关于Python中Yield的基本用法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-10-10