pytorch打印模型結構圖

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

 打印出來的結果是以文本形式顯示，

顯示出模型的每一層是由什么層構成的，一般來說深度卷積網絡是由結構類似的基本模塊組成，內部參數會有區別。

查看模型結構主要是為了看在某些層執行的特別操作。