Is there a script for calculating FLOPs of MSDNet?

[jh-jeong]

Hi, I wonder if there's any code snippet to calculate FLOPs of each classifier in a given MSDNet.
If so, could you share it please?
Thanks.

[taineleau-zz]

Hi,
sorry for the late reply. You may use https://github.com/apaszke/torch-opCounter to calculate the FLOPs of any Lua torch network.

[kazulittlefox]

I have tried to run the snippet above to calculate MSDNet computational cost, but failed
(maybe) because of lack of implementing definition of custom layer or manupilation, like train_joint module .

Anyone else trying running it? And would you like to show how to solve the problem?

Thanks in advance.