今天看文章复习了一下计算量和参数量的计算方法,这里做一下笔记

忽略padding和stride的话,输入层,卷积核和输出层的关系,以及参数量、计算量如下(第二行是一个例子)

![在这里插入图片描述](https://img-blog.csdnimg.cn/20201214182057214.png

关于参数量:每个卷积核有K×K×MK×K×MK×K×M个参数,那么共有N个卷积核,参数量就是K×K×M×NK×K×M×NK×K×M×N

关于计算量:每个卷积核遍历输入层,计算量是K×K×M×Wo×WoK×K×M×W_o×W_oK×K×M×Wo×Wo,那么N个卷积核的计算量就再乘上N。(这里乘上两个WoW_oWo我没有做严格的证明)

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐