kyegomez/VisionMamba

Implementation of Vision Mamba from the paper: "Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model" It's 2.8x faster than DeiT and saves 86.8% GPU memory when performing batch inference to extract features on high-res images

语言：Python · GitHub 星标：497 · Fork：23

话题：ai · machine-learning · mamba · pytorch · recurrent-neural-network · ssm

在 GitHub 上查看原仓库