【发布时间】:2016-11-23 07:33:13
【问题描述】:
通过将底部和顶部 blob 设置为相同,我们可以告诉 Caffe 进行“就地”计算以保持内存消耗。
目前我知道我可以安全地就地使用"BatchNorm"、"Scale" 和"ReLU" 层(如果我错了,请告诉我)。虽然其他层似乎有一些问题(this issue 似乎是一个例子)。
何时在 Caffe 中使用就地层?
它如何与反向传播一起工作?
【问题讨论】:
-
@Shai 谢谢你的回答,肯定有帮助!
标签: machine-learning neural-network deep-learning caffe backpropagation