CUDA:暗网yolov1上的内存不足错误

Question

我试图运行暗网YOLOv1代码上的GPU特斯拉K40。

几天前，makefile中的参数"ARCH= --gpu-体系结构=计算20--gpu-code=COMPU20“一切正常。

但是今天我又做了一次。"make“有效，但在运行YOLO的视频演示时，出现了以下错误

​

我尝试了不同的ARCH参数，比如compute_30，35，52。但什么都没起作用。

可能的解决办法是什么？

注意:暗网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
...

诀窍是使批处理等同于细分。