performance - 深度学习 TensorFlow 模型能否在 Python 中进行训练、腌制并用于 C++ 中的预测？

Question

我已经在 Python 中使用 TensorFlow 学习了一个基于神经网络的模型。

我想将此模型存储在一个文件中，并能够将其加载到 C++ 程序中的内存中以供以后预测。

我正在对我的机器学习模型与用 C++ 编写的标准算法进行比较研究。出于这个原因，我想加载模型并在 C++ 中进行预测，因为我不希望编程语言的内部结构导致实现的运行时差异。

还有其他方法可以使比较保持语言中立吗？

标签： performancetensorflowserializationpickle

Answer 1

是的。我认为你可以使用 bazel（谷歌的 TF 工具）来做到这一点。确保保存检查点后，构建项目。它将创建可执行文件供您在 c++ 中使用。