pytorch打印模型结构图

import torchsummary
from torchvision.models.resnet import *
net = resnet18().cuda()
print(net)

 打印出来的结果是以文本形式显示，

显示出模型的每一层是由什么层构成的，一般来说深度卷积网络是由结构类似的基本模块组成，内部参数会有区别。

查看模型结构主要是为了看在某些层执行的特别操作。

标签：resnet18,torchvision,结构图,打印,pytorch,import,net,模型
来源： https://www.cnblogs.com/ywheunji/p/10608080.html