Pytorch 统计模型参数量的操作 Param.Numel() - 10K编程网_专业编程资源网站

Pytorch 统计模型参数量的操作 Param.Numel()

2021-05-13 240 0 0

param.numel()

返回param中元素的数量

统计模型参数量

num_params = sum(param.numel() for param in net.parameters())
print(num_params)

Python

原文链接: https://blog.csdn.net/weixin_44317740/article/details/104635754

本文链接: https://www.10k.cn/article/19956.html

版权声明: 本文章内容采集或者整合于互联网，仅供学习使用，如若本站内容侵犯了您的合法权益，请及时联系我们删除。

上一篇： Python数据分析入门之教你怎么搭建环境

下一篇： Python爬虫基础之爬虫的分类知识总结