卷积神经网络中参数量的计算与模型占用显存的计算

每次一加载模型GPU的显存就疯狂增长，看了卷积神经网络中参数量的计算与模型占用显存的计算原理才知道缘由。以前还一直怀疑本身写的有bug。感谢https://www.jianshu.com/p/b8d48c99a47c的分享。 下面是我本身结合文章作的一些总结。 占用显存的计算 也就是说咱们的模型占用的显存在不一样阶段的计算方式是不一样的，下面以Vgg16为例。web 1.load model （1

>>阅读原文<<