请问一下，如果显卡的显存不是很大，batch_size只能设置到4，需要采取什么措施吗？

Question

尊敬的Kuan老师，您好，请问一下，如果显卡的显存不是很大，batch_size只能设置到4，需要采取什么措施吗？

期待您的回复！

Answer 1

会写代码的好厨师 2021-01-04 00:09:20

减小backbone的尺寸或者换掉backbone用轻量型网络结构。

• 提问者 慕慕0439991 #1

  “小batch-size+大backbone”跟“大batch-size+小backbone”，哪一种组合最后的效果会更好呢？

• 会写代码的好厨师 回复 提问者 慕慕0439991 #2

  小batchsize一般不要小于16.
  具体那个好，需要根据任务来。比如：文字检测，由于目标比较多，多以batch小一点问题不大。因为一张图片中，目标框会比较多 。
  小的backbone肯定会变差

• 提问者 慕慕0439991 回复 会写代码的好厨师 #3

  好的，谢谢老师！