pytorch 一行代码查看网络参数总量的实现
作者：DX_ByYourself 发布时间：2023-04-23 17:42:36
标签：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))
打印信息如下：
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。
来源：https://blog.csdn.net/CV_YOU/article/details/85002754
0
投稿猜你喜欢
- 前言from collections import namedtuple()命名元祖的工厂函数：在python中，collections 包
本文实例为大家分享了OpenCV基于ORB算法实现角点检测的具体代码，供大家参考，具体内容如下ORB算法是FAST算法和BRIEF算法的结合- 在一些网页应用中，就比如在投票系统中，当我们进行的是多项投票时，我们要求用户最多只能选择几项进行投票，这也是就是说选择复选框的个数最多几个.
- 最近刚学习数据库，首先是了解数据库是什么，数据库、数据表的基本操作，这就面临了一个问题，mysql的安装，我这里下载的是64位的，基于Win
- 本文实例讲述了Python实现的本地文件搜索功能。分享给大家供大家参考，具体如下：偶尔需要搜索指定文件，不想每次都在windows下面去搜索
- 背景上周公司培训了MySQL replication, 这个周末打算用所学来实践操作一下。Master server：MySQL conta
- 前言：为什么要写这篇文章，因为前段时间有一个开源的github中的项目有一个朋友提交了一个pr看了下是帮忙优化了下代码（十分感谢这位网友）。
- 在做维护项目的时，我们经常会遇到索引维护的问题，通过语句，我们就可以判断某个表的索引是否需要重建。 执行一下语句：先分析表的索引 分析表的索
- 一、python线程的模块1.thread和threading模块thread模块提供了基本的线程和锁的支持threading提供了更高级别
- 目录一、为什么要用线程池二、线程池练习演示例子1:使用submit方法演示例子2:使用map方法三、线上数据库测试总结:一、为什么要用线程池
- 前言最近因为工作需要 用selenium做了一个QQ邮箱的爬虫(登录时部分帐号要滑动解锁),先简单记录一下。这个问题先可以分为两个部分:1.
- 前言：本文为 HITwh 网络空间安全专业网络空间安全设计与实践 I 的选题之一，主要实现了远程监控局域网内的主机桌面与网络情况、简单键鼠控
- 上篇文章讲了js中的传值和传址 和 函数的作用域.这章我们来探讨js中的变量，表达式，和运算符 还有一些 js 语句。升级中……1, 表达式
- asp使用session来防止表单多次被提交的方法。formtest.asp' 表单文件<％Randomize&nb
- 你也许已经掌握了id、class、后台选择器这些基本的css选择器。但这远远不是css的全部。下面向大家系统的解析css中30个最常用的选择
- 现在有很多的项目，对计数器的实现甚是随意，比如在实现网站文章点击数的时候，是这么设计数据表的，如：”article_id, article_
- 最简单的php语句把数据库*.sql文件导入数据库 $sql=file_get_contents("text.sql")
- 在完成了应用程序架构和页面的设计后，我们可以开始探讨如何实现多种多样的普通数据到报表展现的任务。前面我们已经了解到如何在从数据访问层和业务逻
- 一、前言提到 limit 优化，大多数 MySQL DBA 都不会陌生，能想到各种应对策略，比如延迟关联，书签式查询等等，之前我也写过一篇优
- vue Print.js打印页面样式不出现解决方案加上这句就好了！完美！vue-print-nb打印问题总结1、表格的列缺失（element
