在gpu上运行train_yolo3.py报out of memory 错误

运行train_yolo3.py
报错:mxnet.base.MXNetError: [17:50:42] src/storage/./pooled_storage_manager.h:157: cudaMalloc failed: out of memory
什么原因呢?求大哥指导啊

“Out of memory” 应该是比较准确的消息。试着减小batch size看一下

cuda显卡内存不足,可以尝试更换显存更大的显卡或者同上面的建议那样减小batch size