Caffe2 中的内存管理

Question

我是 caffe2 的新手，我想知道 caffe2 框架如何将模型存储在 CPU 或 GPU 内存中以及如何维护。我已经尝试阅读它的 github 代码，但到目前为止还没有具体内容。感谢您的帮助:) :)

Answer 1

模型使用 protobuf 存储。和 Caffe 一样。

看看这些：

https://caffe2.ai/docs/tutorial-loading-pre-trained-models.html

https://caffe2.ai/docs/zoo.html

Answer 2

模型存储为运算符图。它在存储中表示为协议缓冲区，并作为 DAG 加载到内存中。