SAM-LST icon indicating copy to clipboard operation
SAM-LST copied to clipboard

GPU memory requirement

Open Raspberry-beans opened this issue 6 months ago • 0 comments

Thanks for the great work!

I will be replicating your approach to fine tune SAM on custom few shot examples. The problem is I have only 8GB GPU memory available, and in the paper it is mentioned that two RTX 3090 graphics cards are used.

Would it be possible to replicate your approach having only 8GB memory space. If not, is there any way I can do fine tuning requiring less than 8GB memory e.g reducing batch_size etc

Your suggestions will be appreciated.

Regards, Muhammad

Raspberry-beans avatar Jan 10 '24 10:01 Raspberry-beans