使用torchsummary打印torch模型結構，包括每層名字以及形狀

import models
from torchsummary import summary

model = models.resnet50(pretrained=False, num_classes=7, scale=1)
print(model)
summary(model, (3, 640, 640))

使用torchsummay模塊，pip install torchsummary

summayr使用方法，加入模型，以及模型輸入tensor的形狀即可，結尾會給出模型的權重大小，反向傳播所需內存大小

使用Print同樣可以直接打印模型結構，兩者各有優劣。