too many values to unpack (expected 3)

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

Closed PlasticBagBand closed 1 month ago

PlasticBagBand commented 1 month ago

作者你好，在我使用ex_vssd.py中的VMAMBA2Block进行一张2d推理的时候，会报这个错误 ![image](https://github.com/user-attachments/assets/0f7d8977-7a7e-4dcf-ae36-f1b7127d089e

请问是怎么回事？

还有一个疑问，在mamba2的forward中直接return u 导致后面的代码运行不到，应该是这样的吗？谢谢

PlasticBagBand commented 1 month ago

报这个错误

Human9000 commented 1 month ago

你好，感谢你的提问。根据问题显示，你输入的数据的shape不正确，正确的格式为 Shape[batch, length, channles]。项目中有提供样例，请阅读 https://github.com/Human9000/nd-Mamba2-torch/blob/main/demo/test_build_net.py

PlasticBagBand commented 1 month ago

成功解决了，谢谢大佬