Model parameters

[liusiping10]

How to calculate the model parameters (3.8M) in the paper? I can't get this result by using thop.

[Zhongkai Zhou]

I also use thop library to calculate the model parameters. For our R-MSFM6 model, the model parameters are 3815044.

[liusiping10]

Sorry, I use sum([param.nelement() for param in model.parameters()])), and get total Prameters = 14821484

[Zhongkai Zhou]

If you use this method above, the parameters of the last two layers of resnet will be included. Indeed, you should use the following method:
from thop import profile
flops, params = profile(model, inputs=(input, )

[liusiping10]

Thanks a lot.