hustvl / Vim

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

Severely low resource efficiency #55

Open yrqUni opened 3 months ago

yrqUni commented 3 months ago

I observed that when using vim/scripts/pt-vim-t.sh for pre-training, the hardware resource efficiency of this code is extremely low. For example, GPU utilization is at 0% most of the time.

hmbb123 commented 3 months ago

I meet the same problem. How can I solve it?