How to get the visible SAM memory consume

pytorch-labs / segment-anything-fast

A batched offline inference oriented version of segment-anything

Apache License 2.0

1.2k stars 71 forks source link

How to get the visible SAM memory consume #107

Closed ZHANG-SH97 closed 11 months ago

ZHANG-SH97 commented 11 months ago

hi, After reading the documentation, I found this picture very interesting, but I did not find a way to draw or generate this image. How can I obtain such an image? Looking forward to your answer :)

cpuhrsch commented 11 months ago

Hello @ZHANG-SH97,

You may find the following snippet useful: https://github.com/pytorch-labs/segment-anything-fast/blob/main/experiments/README.md#memory-profiles

Thank you!