pytorch 计算模型所需算力和参数数量

#!pip install torchstat
import torchvision
from torchstat import stat
model=torchvision.models.resnet18()
model
stat(model=model,input_size=(3,512,512))



...
==================================================================================================================================================================
Total params: 11,689,512
------------------------------------------------------------------------------------------------------------------------------------------------------------------
Total memory: 134.01MB
Total MAdd: 19.01GMAdd
Total Flops: 9.51GFlops
Total MemR+W: 318.6MB