caffe：トレーニング中にベストモデルを保存するにはどうすればいいですか？

フレームワークCaffeでモデルをトレーニングしている間に何度かテストすることができるので、テストの正確性に従ってベストモデルを保存したいのですが、どうすればいいですか？ありがとうございました！caffe：トレーニング中にベストモデルを保存するにはどうすればいいですか？

出典

2017-03-28 hj z

スナップショットの反復回数を確認し、反復回数を同じにします。 –

私はそれについて考えています。すべてのテストフェーズでモデルを保存し、トレーニング後に最良のものを選択します。しかし、私はそれが便利ではないと思うので、CAFFEが最良のものを保存してほしい。現在の精度が以前の精度よりも高くない場合、モデルは保存されません。私はそれがソースコードを修正すべきだと思うが、私は何をすべきか明確ではない。 –

あなたのソルバーファイルには、このような

snapshot: 10000 
snapshot_prefix: "models/bvlc_alexnet/caffe_alexnet_train"

としての行を含める必要があります最初は、どのくらいの頻度のチェックポイントに（保存）カフェモデルパラメータとトレーニングの重みを伝えます。 2番目にはそれらをどこに保存するかが示されます。トレーニングの終了時にも保存されます。ファイルは

caffe_alexnet_train_iter_50000.caffemodel 
caffe_alexnet_train_iter_50000.solverstate

...と表示され、繰り返し50,000で保存されます。最適なセットの名前を変更することもできます。後で実行すると上書きされません。私は通常サブディレクトリをとすることでこれを行い、そこでは最新の結果を移動します。

出典

2017-03-28 15:54:13 Prune

ありがとう、私はこれについても考えていますが、最高のものを保存することができれば、それは可能でしょうか？もちろん、 –

。あなたは完全なプログラムインタフェースを持っています。お使いのOSの機能の中で好きなことをすることができます。それはCaffe計測器の一部ではありません。あなたはそれを独立した拡張として書く必要があります。 – Prune

caffe：トレーニング中にベストモデルを保存するにはどうすればいいですか？

答えて

関連する問題