Tensorflow:用C ++训练模型

时间:2016-08-31 00:33:56

标签: tensorflow

我可以在Tensorflow中用C ++训练模型吗?我没有在它的C ++ API中看到任何优化器。优化器是用Python编写的吗?如果没有,我如何用C ++训练图形?我能够用C ++导入Python训练的图形,但我想用C ++编写代码(训练和推理)

2 个答案:

答案 0 :(得分:1)

我从官方存储库中找到了一个示例培训文件 https://github.com/tensorflow/tensorflow/blob/master/tensorflow/cc/tutorials/example_trainer.cc

我确实认为这只是某种基本训练,并不适用于python API所具有的所有优化。我会继续四处寻找更多信息。

答案 1 :(得分:1)

自动微分目前尚未在C语言中实现张量流,因此在C中训练复杂模型是一项艰巨的任务。他们说他们正在努力:https://github.com/tensorflow/tensorflow/issues/4130