pytorch打印模型结构图

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

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

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

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

上一篇:一个fork()系统调用的问题


下一篇:Linux/Unix