大家好,我发现了一个非常好的Github开源项目,分享给大家,仅使用PyTorch实现的双向Mamba2(BiMamba2)提供N维度支持,包括1d,2d,3d数据的支持,借助BiMamba2,你可以很方便的缝合到任意模型中去提高精度。大家可以理解为这是一个即插即用的Mamba模块,可以轻松嵌入到你现有的模型中,进行创新实验,不需要linux环境,也不需要安装mamba ssm这个函数库,如果您觉得这个项目对您有帮助,可以帮作者点点星星,让更多的人看到它,谢谢你!
www.bilibili.com