detectron2 solver.ims_per_batch和型号之间的差异
在detectron2中,似乎有两个与配置文件中批处理大小相关的设置。
我已经看到他们
SOLVER.IMS_PER_BATCH and MODEL.ROI_HEADS.BATCH_SIZE_PER_IMAGE
既被用作网上批处理大小,也不确定它们有何不同。有什么区别,如果他们一起工作,如何?
In detectron2 there seems to be two settings related to batch size in the config file.
They are
SOLVER.IMS_PER_BATCH and MODEL.ROI_HEADS.BATCH_SIZE_PER_IMAGE
I have seen both being used as batch size online and not sure how they differ. What is the difference and if they work together, how?
如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论
评论(1)
solver.ims_per_batch 是深度学习中通常被称为批处理大小的,它是指在一次迭代中使用的培训示例数量。
model.roi_heads.batch_size_per_image 是
SOLVER.IMS_PER_BATCH is what's commonly known as batch size in deep learning and refers to the number of training examples utilized in one iteration.
MODEL.ROI_HEADS.BATCH_SIZE_PER_IMAGE is a parameter that is used to sample a subset of proposals coming out of RPN to calculate cls and reg loss during training.