张量流会议

Question

我是张量流初学者。所以，请原谅我的问题，如果它是愚蠢的

我检查了使用MNIST数据和tensorflow实现CNN的github代码。 下面的链接：

https://github.com/aymericdamien/TensorFlow-Examples/blob/master/examples/3_NeuralNetworks/convolutional_network.py

不过，我需要保存这段代码生成的模型，但不知道如何做到这一点，因为这代码不涉及使用会话，如何在它上面运行会话？

希望你的回应。

Answer 1

链接代码使用tf.estimator.Estimator来训练模型。其documentation包括如何使用export_savedmodel保存模型。保存的模型可以通过tf.estimator.Estimator初始值的model_dir参数指定其位置来导入。