PyTorch 打印模型

方式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中的创建顺序