预训练模型的参数量如何统计
预训练模型的参数量可以通过以下公式进行统计:
参数量 = 输入通道数 × 卷积核尺寸 × 输出通道数 + 输出通道数
其中,输入通道数指的是每个卷积层的输入通道数;卷积核尺寸指的是卷积核的大小,例如3×3;输出通道数指的是每个卷积层的输出通道数。对于全连接层,其参数量可以通过输入通道数乘以输出通道数来计算。
需要注意的是,预训练模型中还包含了很多其他的参数,例如BN层的均值和方差等,这些参数也需要统计在内。
原文地址: https://www.cveoy.top/t/topic/bnM3 著作权归作者所有。请勿转载和采集!