No module named 'v2dmamba_scan.scan_ops'

[AnChuangZ]

Hi,

I used your Docker to run ../TwoDMamba-main/TwoDVMamba/classification/main.py, but I got the following Traceback:
Traceback (most recent call last): File "/workspace/Code/TwoDMamba-main/TwoDVMamba/classification/models/vmamba.py", line 28, in <module> from .csms6s import selective_scan_fn, selective_scan_flop_jit File "/workspace/Code/TwoDMamba-main/TwoDVMamba/classification/models/csms6s.py", line 38, in <module> raise ImportError("Can not import v2dmamba_scan.") ImportError: Can not import v2dmamba_scan

I have already built the 2DMamba CUDA kernel.
I would like to know whether this is a compilation issue or a missing file issue.

[jingweizhang-xyz]

Did you build the kernel inside the docker? Do you see the compiled .so file? You can refer to https://github.com/cvlab-stonybrook/LBMamba#build-lbmamba-cuda-kernel on how to see if the build is correct.