杰晶网络

PyTorch和Keras计算模型参数的例子
脚本专栏 2024/11/25 佚名
3 2

Pytorch中，变量参数，用numel得到参数数目，累加

def get_parameter_number(net):
  total_num = sum(p.numel() for p in net.parameters())
  trainable_num = sum(p.numel() for p in net.parameters() if p.requires_grad)
  return {'Total': total_num, 'Trainable': trainable_num}

Keras中，直接使用model的summary函数

model = k_model()
model.summary()

以上这篇PyTorch和Keras计算模型参数的例子就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。

标签：
PyTorch,Keras,模型参数
免责声明：本站文章均来自网站采集或用户投稿，网站不提供任何软件下载或自行开发的软件！ 如有用户或公司发现本站内容信息存在侵权行为，请邮件告知！ 858582#qq.com
评论“PyTorch和Keras计算模型参数的例子”
暂无“PyTorch和Keras计算模型参数的例子”评论...
www.imxmx.com 杰晶网络
8,675无损音乐
1,324高清电影
213破解软件
120,141站长资源

更新动态

友情链接

杰晶网络 DDR爱好者之家 南强小屋 黑松山资源网 白云城资源网