ResNet-34で動くけど、-50だとCalling garbage collection 0

429 views
Skip to first unread message

c90

unread,
Jan 25, 2018, 9:54:56 PM1/25/18
to Neural Network Console Users (JP)
付属のResNet-34と-50で、入力サイズと出力を320×320に変更して、連続値推定しております。
ResNet-34ではうまく動きましたが、ResNet-50ではCalling garbage collection 0が3回でて終了してしまいます。※1
Optimaizerなどすべて同じにしましたが、結果はすべて同じ(※1)でした。

そもそも”Calling garbage collection 0”の原因が不明でして、何をすれば前に進めますか?
GPUメモリの不足以前にPC自体のメモリが足りていないということでしょうか?

ちなみにこれまでの引っ掛かりどころは、以下のように対処しました。
 GPUメモリ不足 ⇒ バッチサイズを小さくする
 IndexError: list index out of range ⇒ パス確認(バイトコードが含まれているとか)

よろしくお願いします。

小林由幸

unread,
Jan 30, 2018, 4:05:31 AM1/30/18
to Neural Network Console Users (JP)
こちらのエラーはGPUのメモリ不足が原因と考えられます。

GPUのメモリが不足した場合の基本的な対処方法としては、
CONFIGタブにてBatch Sizeを小さくします。


この時、Batch Sizeを小さくした量に応じてOptimizerのUpdate Intervalと
LR Update Intervalを大きくします(例えばResNet-50でBatch Sizeを32→16の半分にした場合、
Update Intervalはデフォルトの8の倍の16にする)。
これによりBatch Sizeを変える前とほぼ同じ条件での学習が可能になります。
Reply all
Reply to author
Forward
0 new messages