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
投稿猜你喜欢
- 在 MySQL 下，在进行中文模糊检索时，经常会返回一些与之不相关的记录，如查找 "％a％" 时，返回的可能有中文字符，
利用numpy、matplotlib、sympy绘制sigmoid、tanh、ReLU、leaky ReLU、softMax函数起因：深度学- 大家都熟悉迅雷看看里面的电影人气指数这个小图标吧先看看我的效果图再看看迅雷的截图比较好看，是根据电影的人气指数来显示热度，下面我们就来模仿一
- CASSolution使用CAS作为认证协议。A作为主要的认证提供方(provider)。A保留用户系统，其余系统如xxx/www不保留用户
- 多线程类似于同时执行多个不同程序，多线程运行有如下优点：使用线程可以把占据长时间的程序中的任务放到后台去处理。用户界面可以更加吸引人，比如用
- 用window.open打开的窗口中，有时候session变量会丢掉，给asp编程带来的一定的麻烦。用参数传递解决它：<DIV&nbs
- 今天我们实现疲劳检测。 如果眼睛已经闭上了一段时间，我们会认为他们开始打瞌睡并发出警报来唤醒他们并引起他们的注意。我们测试一段视频来展示效果
- 在使用Matplotlib 绘制饼图的时候有些时候一些数据的比列太小在饼图呈现的效果不明显 很容易被覆盖，为了解决这个问题以下就是我个人的心
- 一个不错的js效果，实现了图片预加载，并实时显示图片加载进度。<script> var l=0; var i
- 如果备份的数据库有2个文件，分别是.LDF 和 .MDF，打开企业管理器，在实例上右击---所有任务--附加数据库，然后选择那个.MDF文件
- *注意：本文的内容涉及到修改NTFS磁盘权限和设置安全策略，请务必在确认您了解操作可能的后果之后再动手进行任何的修改。文中提及的权限都是在原
- 导语承载童年的纸飞机你还会叠嘛？如果你是个80后或者90后，那你应该记得小时候玩的纸飞机。叠好后，哈口仙气，飞出去，感觉棒棒哒。虽然是一个极
- 我们在.NET程序的开发过程中，常常需要和用户进行信息交互，比如执行某项操作是否成功，“确定”还是“取消”，以及选择“确定”或“取消”后是否
- pandas处理大数据的限制现在的数据科学比赛提供的数据量越来越大，动不动几十个G，甚至上百G，这就要考验机器性能和数据处理能力。Pytho
- 这篇文章阐述的是一种函数式编程（functional-programming）设计模式，我称之为惰性函数定义（Lazy Function D
- 研究了一段时间酷狗音乐的接口，完美破解了其vip音乐下载方式，想着能更好的追求开源，故写下此篇文章，本文仅供学习参考。虽然没什么
- 如何将产生的密码记录并发送给用户？这里使用了cdonts邮件组件来发送邮件，前提服务器得支持cdonts组件。好了，看看具体实现方法吧，不是
- Python中常用到的两种标准化输入方式：分别sys.stdin和input，两者使用方式大致相同，但是总的来说sys.stdin使用方式更
- Python的优点和缺点本节内容如下：Python的优点Python的缺点使用Python的知名网站Python的优点1. 简单 Pytho
- 最基础的形态学操作有四个，分别是腐蚀、膨胀、开计算和闭计算，`scipy.ndimage分别实现了二值数组和灰度数组的这四种运算二值灰度bi
