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爬虫Requests库的使用详情，文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下2022-08-08
Pandas绝对是Python中处理Excel最快、最好用的库，但是使用 openpyxl 的一些优势是能够轻松地使用样式、条件格式等自定义电子表格，感兴趣的可以了解一下2021-09-09
今天小编就为大家分享一篇python 实现Flask中返回图片流给前端展示，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-01-01
这篇文章主要介绍了python中的字典详细介绍,字典是Python中最强大的数据类型之一,本文讲解了什么是字典、创建字典和给字典赋值 、字典的基本操作、映射类型操作符、映射相关的函数、字典的方法等内容,需要的朋友可以参考下2014-09-09
这篇文章主要为大家详细介绍了如何使用Python内置库实现数据的加密和校验,为开发者提供全方位的数据安全解决方案,感兴趣的小伙伴可以跟随小编一起学习一下2023-12-12
Python Handler处理器和自定义Opener原理详解
这篇文章主要介绍了Python Handler处理器和自定义Opener原理详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-03-03
开发一个程序，用于获取局域网中开启snmp服务的主机ip地址列表，并写入相应文件以便其它程序使用。下面是实现方法2014-02-02
这篇文章主要介绍了python使用json序列化datetime类型实例解析，分享了相关代码示例，小编觉得还是挺不错的，具有一定借鉴价值，需要的朋友可以参考下2018-02-02
这篇文章主要介绍了pycharm 使用anaconda为默认环境的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-02-02
这篇文章主要介绍了Python不规范的日期字符串处理类,可以对一些非正规的日期字符串进行解析、转换、比较等,需要的朋友可以参考下2014-06-06
最新评论