通过summary和ptflops查看model的参数量和计算量

通过summary查询model的参数量：

from torchsummary import summary
    
model = MobileNetV3(n_class=num_class, input_size=224).to("cuda")

    
summary(model,(3,224,224))

通过ptflops查询model的参数量和计算量

from ptflops import get_model_complexity_info

 
flops, params = get_model_complexity_info(model, (3, 224, 224), as_strings=False, print_per_layer_stat=False)
    
print(f'FLOPs: {flops / 1e9:.2f} GFLOPs')
    
print(f'Params: {params / 1e6:.2f} M')