模型参数数量与计算量

原博地址:

https://blog.csdn.net/mzpmzk/article/details/82976871

参数数量(params)与理论计算量(FLOPs)

1.参数数量

      关系到模型大小,单位通常为M,通常参数用float32表示,所以模型大小通常是参数数量的四倍

2.计算公式

      假设卷积核大小为K_{_{h}}*K_{_{w}},输入通道数为C_{in},输出通道数为C_{out},输出特征图的高和宽分别为WH

     CONV卷积层的参数数量与理论计算量为:

      -params: K_{n}*K_{h}*C_{in}*C_{out}

      -FLOPs: K_{n}*K_{h}*C_{in}*C_{out}*H*W,即FLOPs = params*H*W

      

      FC全连接层

      -params:C_{in}*C_{out}

      -FLOPs:C_{in}*C_{out}

     

 

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值