hustvl / Vim

Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model
Apache License 2.0
2.55k stars 159 forks source link

CPU Memory Usage #98

Open leitro opened 3 days ago

leitro commented 3 days ago

Hi! Thanks for this great work, well done!

We are running this code in our project, but the CPU memory has been eaten drastically over 64GB. Have you ever found this problem? Any idea on how to solve it? Thanks!