search

weixuansun / wsss_sam

MIT License
12 stars 0 forks source link

The requirement of GPU #2

Open Hugo-cell111 opened 10 months ago

Hugo-cell111 commented 10 months ago

Hi! I wonder how many GPUs does WSSS of SAM need? Usually 4 or 8 GPUs of V100 is a common setting in semi & weakly supervised learning, and I am a little worried about that WSSS of SAM also needs so many GPUs. Thanks!

weixuansun commented 9 months ago

Hi, thanks for the question. Since we only do inference on SAM, all experiments are ran on a single V10 GPU.

kalpa61 commented 2 months ago

@weixuansun Hello, what is the minimum memory required to train this model? Thank you very much！

weixuansun commented 1 month ago

@weixuansun Hello, what is the minimum memory required to train this model? Thank you very much！

Hi, the whole process does not involve training. So it's only inference memory consumption.