RobNets icon indicating copy to clipboard operation
RobNets copied to clipboard

How to calculate the number of MAC, the number of layers and the model size of the network?

Open wxbbuaa2011 opened this issue 3 years ago • 1 comments

We can use this project with pytorchcount to calculate these parameters? thank you very much!

wxbbuaa2011 avatar Jul 13 '20 05:07 wxbbuaa2011

I have just tried the repo here, and it works fine. Make sure you follow the instruction at README and REMEMBER to copy the corresponding parameters to the function models.robnet(). If anything goes wrong, feel free to let me know.

gmh14 avatar Jul 13 '20 06:07 gmh14