How to compute the model complexity?

NVlabs / stylegan2-ada-pytorch

StyleGAN2-ADA - Official PyTorch implementation

https://arxiv.org/abs/2006.06676

Other

4.13k stars 1.17k forks source link

How to compute the model complexity? #300

Open supersonicMaclaurin opened 9 months ago

supersonicMaclaurin commented 9 months ago

I want to get the parameter number and flops , using officially lib like ptflops, thop, but get the result Warning: module EqualLinear is treated as a zero-op. Warning: module ModulatedConv2d is treated as a zero-op. Warning: module FusedLeakyReLU is treated as a zero-op. Warning: module StyleConv is treated as a zero-op.