2017-09-17 10 views
0

私はテンソルフローの初心者です。だから私の疑問を愚かにしてください。テンソルフローのセッション

MNISTデータとテンソルフローを使ってCNNを実装するためのgithubコードをチェックしました。 下記のリンク:

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

しかし、私はこのコードによって生成されたモデルを保存する必要がありますが、このコードは、セッションの使用を伴わないとしてどのように、どのように行うのか分かりませんそれに対するセッションが不完全ですか?

ご回答ありがとうございます。

答えて

0

リンクコードはtf.estimator.Estimatorを使用してモデルをトレーニングしています。そのdocumentationには、export_savedmodelを使用してモデルを保存する方法が含まれています。保存されているモデルは、イニシャライザの引数model_dirで指定してインポートできます。

+0

チェックポイントを保存しましたが、チェックポイントの数(チェックポイントを保存するまでの繰り返し回数)を指定する方法が不明です。また、別の質問をして申し訳ありません。私はそのコードに予測されたクラスを出力したいと思う。セッションなしでどうすればいいですか? – Loura

関連する問題