pytorch查看模型的参数总量、占用显存量以及flops

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

>>阅读原文<<