caffe内存需要怎么计算?

时间:2016-12-05 12:50:45

标签: deep-learning caffe

当您想要训练网时,您将获得如下日志信息:

current

你如何解释这个数字?它是以字节为单位的吗?

1 个答案:

答案 0 :(得分:0)

要回答直接问题,图层的内存使用量以字节为单位,而不是位。

总的来说,报告的内存是可以在初始化时计算的所有层所需的内存。参数和缓冲区空间是根据需要分配的,因此以汇总和报告它们没有用 - 由于序列化重用,它们中的一些可能会占用更少的总内存。

正如一个答案所说,确定最大内存使用量的最有效方法是运行前向传球并抓住其高水位线。