在detectron2中,配置文件中似乎有两个与批处理大小相关的设置。
他们是
SOLVER.IMS_PER_BATCH and MODEL.ROI_HEADS.BATCH_SIZE_PER_IMAGE
我已经看到,两者都被用作批次大小在线,并不确定它们有何不同。有什么区别,如果他们一起工作,怎么做?
发布于 2022-08-01 16:01:38
SOLVER.IMS_PER_BATCH通常被称为深度学习中的批处理大小,它指的是在一次迭代中使用的训练示例的数量。
MODEL.ROI_HEADS.BATCH_SIZE_PER_IMAGE是一个参数,用于对来自RPN的提案子集进行采样,以计算训练期间的cls和reg损失。
https://stackoverflow.com/questions/72637875
复制相似问题