import torchsummary
from torchvision.models.resnet import *
net = resnet18().cuda()
print(net)
打印出来的结果是以文本形式显示,
显示出模型的每一层是由什么层构成的,一般来说深度卷积网络是由结构类似的基本模块组成,内部参数会有区别。
查看模型结构主要是为了看在某些层执行的特别操作。
2023-10-08 16:09:58
import torchsummary
from torchvision.models.resnet import *
net = resnet18().cuda()
print(net)
打印出来的结果是以文本形式显示,
显示出模型的每一层是由什么层构成的,一般来说深度卷积网络是由结构类似的基本模块组成,内部参数会有区别。
查看模型结构主要是为了看在某些层执行的特别操作。
下一篇:Linux/Unix