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;
191 stars 7 forks source link

您好，可以问一下您的环境的具体配置吗 #10

Open Dim-1012 opened 2 months ago

Dim-1012 commented 2 months ago

感谢

Human9000 commented 1 month ago

感谢你的提问，下面是回答： 本项目使用的环境是torch2，具体的版本没有很强的依赖性 我这边使用的版本是： torch==2.1.2+cu118