Sigkill错误:在自定义数据集上训练Mask RCNN模型

Question

我正在尝试在自定义数据集上训练Mask-RCNN模型。但是，几秒钟后，我的程序停止并返回“退出代码137 (被信号9 sigkill中断)”错误。

Answer 1

当我们的模型使用了过多的内存时，就会出现这个错误。设置小批量会有所帮助。