pytorch 一行代码查看网络参数总量的实现

更新时间：2021年05月12日 11:21:00 作者：DX_ByYourself

这篇文章主要介绍了pytorch实现一行代码查看网络参数总量的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

大家还是直接看代码吧~

netG = Generator()
print('# generator parameters:', sum(param.numel() for param in netG.parameters()))
netD = Discriminator()
print('# discriminator parameters:', sum(param.numel() for param in netD.parameters()))

补充：PyTorch查看网络模型的参数量PARAMS和FLOPS等

在PyTorch中，可以使用torchstat这个库来查看网络模型的一些信息，包括总的参数量params、MAdd、显卡内存占用量和FLOPs等。

示例代码如下：

from torchstat import stat
from torchvision.models import resnet50, resnet101, resnet152, resnext101_32x8d
model = resnet50()
stat(model, (3, 224, 224))

打印信息如下：

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。

您可能感兴趣的文章:

python使用pycharm环境调用opencv库
这篇文章主要介绍了python使用pycharm环境调用opencv库，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2018-02-02
python输出电脑上所有的串口名的方法
今天小编就为大家分享一篇python输出电脑上所有的串口名的方法，具有好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-07-07
如何利用Python保留指定位数的小数
这篇文章主要给大家介绍了关于如何利用Python保留指定位数的小数,,代码简单易懂,非常不错,对大家学习或者使用Python具有一定的参考借鉴价值,需要的朋友可以参考下
2023-08-08
python面向对象版学生信息管理系统
这篇文章主要为大家详细介绍了python面向对象版学生信息管理系统，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-06-06
通过Python中的http.server搭建文件上传下载服务功能
通过本文我们学习了如何使用Python的http.server模块搭建一个基本的HTTP服务器,并实现文件下载服务,介绍了如何设置服务器端口、自定义文件目录、定制HTTP响应头以及处理GET请求,感兴趣的朋友跟随小编一起看看吧
2024-08-08
如何从csv文件构建Tensorflow的数据集
这篇文章主要介绍了如何从csv文件构建Tensorflow的数据集，帮助大家更好的理解和使用Tensorflow，感兴趣的朋友可以了解下
2020-09-09
python实现简单的井字棋游戏(gui界面)
这篇文章主要介绍了python如何实现简单的井字棋游戏，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下
2021-01-01
Atom Python 配置Python3 解释器的方法
今天小编就为大家分享一篇Atom Python 配置Python3 解释器的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-08-08
Django+RestFramework API接口及接口文档并返回json数据操作
这篇文章主要介绍了Django+RestFramework API接口及接口文档并返回json数据操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-07-07
浅谈python新手中常见的疑惑及解答
下面小编就为大家带来一篇浅谈python新手中常见的疑惑及解答。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2016-06-06