为什么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)
谢谢!
答案 0 :(得分:0)
TensorFlow tf.layers
tutorial使用tf.estimator.Estimator
作为高级API,隐藏构建会话的详细信息,并编写检查模型和日志摘要的训练循环。而是指定一个描述输入数据的input_fn
和一个描述图层结构的model_fn
。
如果您希望直接使用tf.Session
(或tf.train.MonitoredSession
)API,则可以直接在自己的代码中调用model_fn
,并根据需要创建优化程序,保护程序等