2017-08-25 48 views
0

爲什麼Tensorflow圖層教程中沒有Session對象?是否有可能以某種方式獲得它?未在Tensorflow MNIST教程中指定會話對象

教程: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

TensorFlow tf.layers tutorial使用tf.estimator.Estimator作爲隱藏構建會話的詳細信息的高級API,並編寫一個訓練循環來檢查您的模型並記錄摘要。而是指定描述輸入數據的input_fn和描述層結構的model_fn

如果你更喜歡直接使用tf.Session(或tf.train.MonitoredSession)API,你可以直接在自己的代碼調用model_fn並根據需要創建一個優化,金丹等。

+0

謝謝你的回答。我的問題涉及到保存模型,但它似乎是自動使用'Estimator'類來完成的,正如您所指出的那樣。稍後加載它,事實證明我只需要定義它:'learn.Estimator(model_fn = cnn_model_fn,model_dir =「./ model」)'就是這樣。 – user