nnUNet icon indicating copy to clipboard operation
nnUNet copied to clipboard

How to calculate the num of parameters and MACs(Multiply-Accumulate Operations)?

Open Tunaaaaa opened this issue 2 years ago • 1 comments

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~

Tunaaaaa avatar Aug 13 '22 16:08 Tunaaaaa

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

FabianIsensee avatar Aug 23 '22 13:08 FabianIsensee

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.

GregorKoehler avatar Jul 31 '23 13:07 GregorKoehler