Inaccurate number of parameters from torchsummary

[Thity]

The number of parameters reported for each FixCaps model seems to be coming from torchsummary (see cell. 6 in FixCaps-DS notebook).

However, torchsummary parameter count is inaccurate for custom modules, in particular when parameters name is not "weight" or "bias" (see this issue). The count does not consider network.digits.W parameters that represent the majority of the FixCaps-DS model's parameters.