用GPU跑yolov3模型时,出现下面错误
layer filters size input output
0 conv 16 3 x 3 / 1 224 x 224 x 3 -> 224 x 224 x 16 0.043 BFLOPs
1 CUDA Error: out of memory
darknet: ./src/cuda.c:36: check_error: Assertion `0' failed.
Aborted (core dumped)
尝试解决方法:
所使用的模型cfg文件中的subdivision的参数。由subdivisions=8改成subdivisions=64。