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

mamba2中n_layer参数 #14

Open GZLiao-sysu opened 1 month ago

GZLiao-sysu commented 1 month ago

请问在mamba2中n_layer参数起到什么作用吗？我似乎没有在后面的代码中看到调用这个参数

Human9000 commented 1 month ago

感谢你的提问 n_layer是mamba2的模型参数，不是模块参数，它控制了一个mamba2的模型中，有多少个mamba2block的模块。该项目是面向mamba2block的模块进行的工作，所有该项目中应该很少出现n_layer这个参数