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

pretrained weight #12

Open ppbangKGT opened 1 month ago

ppbangKGT commented 1 month ago

您好！感谢您出色的工作！ 请问您是否有在ImageNet上训练过该模型？ 如果有的话，不知道您是否方便将训练的权重分享？因为希望使用mamba2作为图像的特征提取器，如果没有一个预训练好的权重，担心模型收敛较慢。 谢谢！

Human9000 commented 1 month ago

感谢你的提问 该项目并没有在imageNet上做过训练，因为它仅仅是一个block模块，如果你需要预训练的模型，可以阅读vssd（https://github.com/YuHengsss/VSSD）的相关项目。