私はcaffeのデータにfcnを実行しようとしていました。イメージセットをlmdb
にconvert_imageset
で組み込みました。私はnet
を訓練したかったしかし、一度、それは私に次のエラーを与えた:convert_imageset.cppの一括してイメージ数を減らしてGPUのメモリ不足を解消することはできますか?
Check failed: error == cudaSuccess (2 vs. 0) out of memory
*** Check failure stack trace: ***
.....
Aborted (core dumped)
私はメモリ障害を解決するために、多くのオンラインリソースを経て、それらのほとんどは、バッチサイズを減らすことを示唆しています。さらに、画像のサイズを256x256に縮小しました。私はまだこの問題に取り組むことができませんでした。 このコマンドでGPUのメモリを確認しましたnvidia-smi
、モデルはNvidia GT 730
、メモリは1998 MiB
です。 train_val.prototxt
のバッチサイズは1なので、train_val.prototxt
には何もできません。だから、私の質問は以下のとおりです。ターミナルでログファイルを見ることによって
- が、私は
convert_imageset
がLMDBにデータを変換する時はいつでも、それはグループで1000年画像を取っていることに気づきました。この数値を143
と151
のconvert_imageset.cpp
に変更することは可能ですか?convert_imagesetを使用してcaffeを再コンパイルしてイメージをlmdbに変換することはできますか?それは理にかなっていますか? - 1の質問への答えは、私は再びカフェをコンパイルする方法、 私は
build
フォルダを削除して、もう一度 最初からカフェのインストールを行う必要があり、そうであるならば? - どのようにcaffeがLMDBデータを処理しますか? convert_imagenetを実行している間に1000個の画像を一括して表示するのは好きですか?
本当にありがとうございます。 おかげで...
ありがとうございました。はい、バッチサイズが1のFCNセマンティックセグメンテーションです。私の問題の解決策はありますか? –
@ S.EBは、より大きなメモリを備えた新しいGPUを購入します。 :| – Shai
ご協力いただきありがとうございます –