在Tensorflow MNIST教程中未指定会话对象

时间:2017-08-25 12:09:37

标签: python tensorflow mnist

为什么Tensorflow图层教程中没有Session对象?是否有可能以某种方式获得它?

教程:https://www.tensorflow.org/tutorials/layers

源代码:https://github.com/tensorflow/tensorflow/blob/r1.3/tensorflow/examples/tutorials/layers/cnn_mnist.py

在进一步开发中,可能需要会话对象来保存训练模型,例如:

session = tf.Session()
saver = tf.train.Saver()

# some processing here

saver.save(session, 'myModel',global_step=1000)

谢谢!

1 个答案:

答案 0 :(得分:0)

TensorFlow tf.layers tutorial使用tf.estimator.Estimator作为高级API,隐藏构建会话的详细信息,并编写检查模型和日志摘要的训练循环。而是指定一个描述输入数据的input_fn和一个描述图层结构的model_fn

如果您希望直接使用tf.Session(或tf.train.MonitoredSession)API,则可以直接在自己的代码中调用model_fn,并根据需要创建优化程序,保护程序等