并行速度变慢

您好，请问您文章使用 的是ss2D型vss还是mamba呢，请问您这种方法适用于SS2D型的vss吗，我把vmunet中的vss改成并行，虽然参数和Gflops下降了，但是训练速度却变慢了，甚至会出现想显存爆炸的情况，这是为什么呢