When the last layer of the fully connected layer becomes CapProNet

[ghost]

When the last layer of the fully connected layer becomes CapProNet, how do you set up the initialization of the network? Or just change the CapProNet part, do not change the initialization of the network? thanks.