2017-01-25 11 views
0

私の目標は、テンソルフローモデルを保存し、それを復元することですが、形状変数を変えることです(新しいクラスと新しいトレーニング例を追加したい)。 this questionでは、シェイプの要素の総数が同じで、新しいグラフを作成する必要がある場合、Saverのオプションの変形は機能すると説明されています。 私の質問です:古い訓練されたモデルで新しいグラフを作成することは可能ですか?それで、新しいクラスでトレーニングを再開します。テンソルフローの前のモデルから新しいグラフを作成するにはどうすればよいですか?

答えて

1

モデルの共有部分を古いチェックポイントから初期化できます。見てください:tf.contrib.framework.init_from_checkpoint

+0

ありがとう、それは正常に動作します。私は公式のドキュメント[リンク](https://www.tensorflow.org/versions/master/api_docs/python/contrib.framework/checkpoint_utilities#init_from_checkpoint)を使用しました。 – emerjay

関連する問題