方式1
print 打印网络的层
import torchsummary
from torchvision.models.resnet import *
net = resnet18().cuda()
print(net)
方式2
torchsummary.summary
net = resnet18()
torchsummary.summary(net, (3, 224, 224))
总结
两种方法打印出来的都是net函数中init中的层， 定义到forward中的层无法显示， 并且层的先后顺序是init中的创建顺序