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中进度条tqdm包使用方法及特性,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-08-08
这篇文章主要介绍了 python基础教程之匿名函数lambda的相关资料,需要的朋友可以参考下2017-01-01
这篇文章主要给大家介绍了关于python中os和sys模块的区别与常用方法的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧。2017-11-11
在日常编程中,我们经常会遇到需要将二进制文件转换为文本文件的情况,在Python中,我们可以利用各种库和技术来完成这项任务,本文将介绍如何使用Python将二进制文件转换为文本文件,并提供实用的代码示例,需要的朋友可以参考下2024-04-04
这篇文章主要介绍了python scrapy脚本报错问题及解决,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-02-02
下面小编就为大家带来一篇解决Python 遍历字典时删除元素报异常的问题。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2016-09-09
这篇文章主要介绍了Pandas库之DataFrame使用的学习笔记，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-06-06
在我们使用Python语言进行机器学习编程的时候，这是一个非常常用的基础库。本文针对Python 机器学习库 NumPy入门教程，感兴趣的朋友跟随脚本之家小编一起学习吧2018-04-04
这篇文章主要介绍了python 列表推导和生成器表达式的使用方法，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下2021-02-02
这篇文章主要介绍了python学习之读取配置文件，文章基于python的相关资料展开对主题的详细介绍，具有一定的参考价值，需要的小伙伴可以参考一下2022-04-04
最新评论