import torch
from torchstat import stat
import torchvision.models as models
net = models.vgg11()
stat(net,(3,224,224)) # (3,224,224)表示输入图片的尺寸
使用 torchstat库
结果为:
import torch
from torchstat import stat
import torchvision.models as models
net = models.vgg11()
stat(net,(3,224,224)) # (3,224,224)表示输入图片的尺寸
使用 torchstat库
结果为: