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

忽略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

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐