c++ - 急切加载整个模型以估计 Tensorflow Serving 的内存消耗

标签 c++ tensorflow machine-learning memory tensorflow-serving

随着预测的执行,Tensorflow Serving 惰性初始化模型 DAG 中的节点。这使得很难估计保存整个模型所需的内存 (RAM)。 是否有标准方法强制 Tensorflow Serving 将模型完全初始化/加载到内存中?

最佳答案

您可以使用模型预热来强制将所有组件加载到内存中。 [1]

[1] https://www.tensorflow.org/tfx/serving/saved_model_warmup

关于c++ - 急切加载整个模型以估计 Tensorflow Serving 的内存消耗,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56012618/

相关文章:

c++ - 使用 BitBlt 获取另一个窗口的单个像素

c++ - 巴泽尔 |如何将资源复制到构建目录?

tensorflow - 导入错误 : libcudnn when running a TensorFlow program

从头开始实现的 C++ 神经网络在 MNIST 上无法达到 50% 以上

python - 如何为 sklearn CountVectorizer 设置自定义停用词?

machine-learning - 决策树学习和杂质

C++ - 使用基类作为模板参数

c++ - std::regex 不识别 $

sql - DeepMind如何减少Atari游戏Q值的计算?

Tensorflow: `tf.nn.sparse_softmax_cross_entropy_with_logits` 中应用的确切公式是什么?