When i add a new layer to VSSM, the training process has the "Parameter indices" error?

[zhangye-zoe]

I want to add a new network layer to class VSSM(nn.Module), like following:

But when training the network, which has the following error:

Could you give me some advice to solve the problem, thank you!

[MzeroMiko]

Please ensure that all the parameters are involved when calculating loss, otherwise, some parameters may not recieve gradients.