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

 

更多推荐