Guidance_on_avoid_using_useless_network

管他呢，先安这玩意

> pip install thop

然后引入

from thop import profile


model_category = XXnet()
input_test = torch.randn(1, 3, 256, 256)
flops, params = profile(model_category, (input_test,))