Open Thity opened 1 year ago
The reported parameter count for CapsNet is obtained from torchsummary. However, it's worth noting that torchsummary's parameter count can be inaccurate when dealing with custom modules. In this case, it doesn't account for parameters in 'network.digits.W,' which amount to approximately 0.51 million, representing roughly 26% of CapsNet's total parameters. Thank you for your attention.
The number of parameters reported for CapsNet seems to be coming from torchsummary.
However, torchsummary parameter count is inaccurate for custom modules, in particular when parameters name is not "weight" or "bias" (see https://github.com/sksq96/pytorch-summary/issues/65). The count does not consider network.digits.W parameters that represent ~25% of CapsNet parameters.