search

Human9000 / nd-Mamba2-torch

Only implemented through torch: "bi - mamba2" , "vision- mamba2 -torch". support 1d/2d/3d/nd and support export by jit.script/onnx;
189 stars 6 forks source link

加到其他模型中 #21

Open xiaoze0801 opened 2 days ago

xiaoze0801 commented 2 days ago

你好，我想问一下，这个mamba2模块，我直接加到其他模型中就可以用吗？比如unet模型，我想用mamba2代替卷积快，是不是直接用你给的代码替换就可以呀

Human9000 commented 7 hours ago

是的，可以直接加载的