search

pytorch-labs / segment-anything-fast

A batched offline inference oriented version of segment-anything
Apache License 2.0
1.21k stars 72 forks source link

requires gpu large memory #125

Open RohanChacko opened 5 months ago

RohanChacko commented 5 months ago

Hi, This model takes up ~23gb of vram on my rtx 3090. I cannot run the model for more than two images before I hit OOM. Vanilla SAM only takes up ~5gb of vram. Is this expected behavior? Thanks!