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

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

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

SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。

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

 

扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄