keras-retinanet 报内存不足问题

keras-retinanet 使用训练32×32小图片。硬件8核,内存16G,显卡1060。

没有参数执行报错如下:

OP_REQUIRES failed at gather_nd_op.cc:47 : Resource exhausted: OOM when allocating tensor with shap

由于内存不足引起,应该是被拉伸了。增加–no-resize参数问题解决。

发表评论

电子邮件地址不会被公开。 必填项已用*标注