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

Question

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

Answer 1

当我们的模型使用过多内存时会发生此错误。设置小批量会有所帮助。