pytorch查看模型的參數總量、佔用顯存量以及flops

from torchstat import stat import torchvision.models as models model = models.resnet152() stat(model, (3, 224, 224))  

>>阅读原文<<