IRMVLab/Point-Mamba

TypeError: Mamba.__init__() got an unexpected keyword argument 'bimamba_type'

Opened this issue · 0 comments

I write a demo in PointMamba.py as follows,

if name == 'main':
model = PointMamba(
in_channels = 2,
channels = [96, 192, 384, 384],
num_blocks = [2, 2, 18, 2],
# num_heads: List[int] = [6, 12, 24, 24],
drop_path = 0.5,
nempty = True, stem_down = 2
)
# print(model)
x = torch.randn((1024, 3))
print(x.shape)
y = model(x)
print(y.shape)

And a problem happened:
TypeError: Mamba.__init__() got an unexpected keyword argument 'bimamba_type'

How to solve it?
I request for a help, thank you.