How to calcalate the Trainable params in Mona?

[aulaywang]

Hi author, it seems that the trainable params in Mona is hard to calculate. We have tried torchsummary, but it didn't work.
For example, we use torchsummary to calculate the Swin Transformer Tiny and the result is:

Total params: 28,265,032
Trainable params: 28,265,032
Non-trainable params: 0
Input size (MB): 0.57
Forward/backward pass size (MB): 252.99
Params size (MB): 107.82
Estimated Total Size (MB): 361.39

We hope the Mona can add the similar function for precise comparison. Thanks!