Tensorflow r1.3 estimator.DNNClassifier保存并加载图

时间:2017-08-23 10:17:50

标签: python c++ tensorflow

我正在使用最新的Tensorflow版本1.3.0并通过以下方式创建了一个DNNClassifier:

classifier = tf.estimator.DNNClassifier(...)

并在python中训练它。

如何在c ++中保存和重用图形/模型? 我只找到加载tf.contrib.learn.DNNClassifier的教程,这不是我想做的...我知道检查点是自动保存的,但是我想加载创建的图形以在c ++中进行预测,并且仅自动保存的文件为graph.pbtxt(除了许多events.outmodel.ckpt个文件外)。我可以像graph.pb一样加载该文件吗?

0 个答案:

没有答案