pytorch 计算模型所需算力和参数数量
2023-01-15 本文已影响0人
午字横
#!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
