python - 使用faster_rcnn_nas_coco模型训练时设置 "second_stage_batch_size*"是什么意思？

Question

I used Object detection API to train faster_rcnn_nas model.

但是我得到了“内存不足”的错误，但是当我添加

second_stage_batch_size: 4

上面这行代码解决了我的问题，我现在正在训练。但是我想知道上面这行是什么意思？

Here是上面一行的定义。

但是我不是很明白。谁能帮我解释一下？

Answer 1

如 here 中所述，“batch_size”主要是术语，意思是对训练过程有贡献的proposals的数量，即那些用于计算分类和边界框回归并且对损失有贡献的proposals。因此，如果这些东西太多，您的 GPU 就会内存不足，这是有道理的。