pytorch 一行代码查看网络参数总量的实现
作者:DX_ByYourself 时间:2023-04-23 17:42:36
大家还是直接看代码吧~
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
标签:pytorch,查看,网络参数
0
投稿
猜你喜欢
php字符串截取函数用法分析
2023-06-28 22:19:26
对python实时得到鼠标位置的示例讲解
2022-02-21 10:01:25
服务器XMLHTTP(Server XMLHTTP in ASP)基础知识
2011-03-06 11:11:00
PyCharm永久激活方式(推荐)
2023-06-12 00:53:33
asp HTTP_X_FORWARDED_FOR和REMOTE_ADDR
2010-07-02 12:33:00
微软的jQuery国际化插件
2010-07-02 12:46:00
让字体美起来
2011-06-14 09:50:21
了解WEB页面工具语言XML(三)支持工具
2008-09-05 17:18:00
python3安装pip3(install pip3 for python 3.x)
2023-03-30 07:19:09
MySQL数据库root权限丢失解决方案
2008-07-13 13:59:00
asp如何做一个分页程序?
2010-07-11 21:13:00
Python实现将绝对URL替换成相对URL的方法
2023-08-28 13:40:08
python函数形参用法实例分析
2023-09-08 21:07:09
本地机apache配置基于域名的虚拟主机详解
2023-11-17 07:40:37
Python中re模块:匹配开头/结尾(^/$)
2021-08-31 02:29:13
使用python实现希尔、计数、基数基础排序的代码
2023-07-12 09:02:24
改善登陆界面的用户体验: 自动聚焦表单
2009-12-09 16:13:00
Python字符串的索引与切片
2021-08-17 23:46:00
asp OpenTextFile文本读取与写入实例代码
2011-04-15 11:00:00
python基础while循环及if判断的实例讲解
2021-02-18 06:56:06