The model uses the Mamba module instead of the VSS module.

MrBlankness / LightM-UNet

Pytorch implementation of "LightM-UNet: Mamba Assists in Lightweight UNet for Medical Image Segmentation"

https://arxiv.org/abs/2403.05246

Apache License 2.0

279 stars 25 forks source link

The model uses the Mamba module instead of the VSS module. #31

Open DongdongMeng opened 3 months ago

DongdongMeng commented 3 months ago

The code for LightMUNet seems to be incorrect. You are using the same Mamba module as Umamba, instead of the VSS module. Please check the class MambaLayer(nn.Module) and respond to this, thank you.

zs1314 commented 2 months ago

Yes!I think you are wrong!Please check the and respond to this, thank you.