[BUG] Cannot import name 'SSM' from 'zeta.nn'
9
👍
0Describe the bug
The zeta module which is a required import does not have the attribute ssm
To Reproduce
Steps to reproduce the behavior:
- Go to 'vision_mamba/model.py'
- Click on 'model.py'
- If you run example.py in the root directory, you will get an import Error:
ImportError: cannot import name 'SSM' from 'zeta.nn' (C:\Users\mseb\miniconda3\envs\py38\lib\site-packages\zeta\nn_init_.py)
Expected behavior
We would expect the import to work
Additional context
I also tried 'from zeta.nn.modules import SSM'. This also does not work although the documentation here [https://zeta.apac.ai/en/latest/zeta/nn/modules/ssm/] suggests that it should work.
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
385
Contributors get 40% of received funds after fees
How does funding with Polar work?
1
Pay now to fund the work behind this issue.
2
Get updates on progress being made.
3
Maintainer is rewarded once the issue is completed.
FAQ
Backer
You're funding impactful open source efforts
Contributor
You want to contribute to this effort
Maintainer
You want to get funding like this too