search

HarborYuan / ovsam

[ECCV 2024] The official code of paper "Open-Vocabulary SAM".
https://www.mmlab-ntu.com/project/ovsam
Other
945 stars 28 forks source link

Reduce Train Batch Size #19

Closed WuHenry-0609 closed 9 months ago

WuHenry-0609 commented 10 months ago

Hi! I want to reduce train batch size from 2 to 1. How can I do it? I'm looking forward to your early reply. Thanks!

HarborYuan commented 9 months ago

Hi @WuHenry-0609 ,

You can change the batchsize on each GPU from 2 to 1 (here).