search

ge-xing / SegMamba

SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentation
https://arxiv.org/abs/2401.13560
316 stars 16 forks source link

light_traing #28

Open smanman opened 3 months ago

smanman commented 3 months ago

Hello, I would like to ask you what is the matter with light_training, I am doing 3D medical image segmentation, using A40 training, using two SegMamba as my backbone segmentation network, bs=4 burst memory, bs=2 I do not know if I can achieve the optimal results, can there be a lightweight mamba can be used?

ge-xing commented 3 months ago

I think bs cannot affect the final performance.