2016-06-20 10 views
0

TensorFlowのN回目の繰り返しごとにモデルのチェックポイントを設定する(つまりスナップショットを保存する)ベストプラクティスは何ですか?繰り返しごとにモデルのチェックポイントを設定する

もちろん、カスタムロジックを使用してtf.train.Saver.saveと呼ぶこともできますが、その内部メカニズムがあるのだろうかと疑問に思っていましたか?

答えて

0

私は、チェックポイントを保存する頻度に基づいて、このような何かを:

if (epoch % N) == (N - 1): 
    saveto=CkptDir+"ckpt/model"+str(N)+".ckpt" 
    save_path = saverRun.save(session, saveto) 

これは、これまでにチェックポイントファイルNエポックのが保存されます。

0

解決策を見つけた。 max_to_keepkeep_checkpoint_every_n_hoursを設定して保存することができます。

関連する問題