NVlabs / stylegan2-ada-pytorch

StyleGAN2-ADA - Official PyTorch implementation
https://arxiv.org/abs/2006.06676
Other
4.09k stars 1.16k forks source link

How to compute the model complexity? #300

Open supersonicMaclaurin opened 8 months ago

supersonicMaclaurin commented 8 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.