2017-08-25 10 views
0

Tensorflowレイヤーチュートリアルにセッションオブジェクトがないのはなぜですか?それは何らかの形でそれを得ることは可能ですか?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

は、セッションの構築の詳細を隠す高レベルAPIとしてtf.estimator.Estimatorを使用し、モデルとログの要約をチェックポイントするトレーニングループを作成します。代わりに、入力データを記述する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

関連する問題