CUDA：darknet yolov1のメモリ不足エラー

Question

GPU Tesla K40でdarknet YOLOv1コードを実行しようとしています。

メイクファイルで、パラメータ "ARCH = --gpu-architecture = compute_20 --gpu-code = compute_20"で数日前にすべてうまく動作していました。

しかし、今日私は再び作った。 "make"は動作しましたが、ビデオでYOLOのデモを実行すると、次のエラーが発生します。

私はcompute_30,35,52などの異なるARCHパラメータを試しました。

可能な解決策は何でしょうか？

注：darknet yoloV2は問題なく動作しています。私は次のブログを実装しています：http://guanghan.info/blog/en/my-works/train-yolo/

Answer 1

cfg/tiny-yolo-voc.cfgを修正しようとしましたか？

それは次のようになります。

[net] 
batch=64 
subdivisions=64 
width=416 
height=416 
channels=3 
momentum=0.9 
decay=0.0005 
angle=0 
saturation = 1.5 
exposure = 1.5 
hue=.1 
... 

トリックは細分化にバッチが等しくなっています。