2017-01-04 1 views
3

モデルは10回ごとに保存されています。だから、私は私の保存されたディレクトリに次のファイルがあります。特定の繰り返しまたはチェックポイントでモデルをテンソルフローにロード/復元する

checkpoint model-50.data-00000-of-00001 model-50.index model-50.meta 
model-60.data-00000-of-00001 model-60.index model-60.meta 

など100まで。モデル50のみをロードする必要があります。私は の70回の反復後のNaN値を得ているからです。 deafaultによって、私がセーバーを復元しているときに、最終的なチェックポイントが検索されます。ですから、どのようにモデル50を具体的にロードすることができますか?助けてください、そうでなければ、私はモデルの利益を最初から実行しなければなりません、それは時間がかかります。

+0

は、私はあなたが手動で非常に粗製のハックとして – martianwars

+0

@martianwarsを 'checkpoint'を編集することができると思います - あなたはそれをより具体的なことはできますか? –

+0

私は保存ディレクトリのファイル 'checkpoint'を意味していました。 – martianwars

答えて

3

tf.train.Saverの関数restore()を使用しているので、last_checkpoints関数を使用して、使用可能なすべてのチェックポイントのリストを取得できます。このリストにはmodel-50model-60の両方が表示されます。

正しいモデルを選んで、このようなrestore()に直接渡し

saver.restore(sess, ckpt_path) 
関連する問題