神经网络模型大小计算

来源:互联网 发布:php管理系统源代码 编辑:程序博客网 时间:2024/05/22 11:32

对于卷积层,若输入为20维,输出为50个feature map,则该层的权重矩阵w大小为[5,5,20,50],因此参数个数为:$

$

由于w为float类型,占用4bytes,因此字节数为:

原创粉丝点击