nnUNet
nnUNet copied to clipboard
How to calculate the num of parameters and MACs(Multiply-Accumulate Operations)?
I'd like to know num of parameters and MACs for full-res network and cascade network.
Usually I use:
from thop import profile
macs, params = profile(net, inputs=(input,))
for this purpose.
Is there any easy method to do this in nnunet's achitecture? Thank you very much~
Hi, unfortunately there is no easier way to do this. Just integrate this piece of code somehwere in nnU-Net, for example in the code that plots the architecture using hiddenlayer Best, Fabian
Hi @Tunaaaaa, Did this issue get resolved? If you still encounter problems, please let us know! Otherwise, we'll close this issue in roughly 2 weeks.