import torch
import torchvision
!pip install torchsummary
import torchsummary
net = torchvision.models.vgg19()
查看模型结构
print(net)
参数量、各层输入输出矩阵大小
torchsummary.summary(model=net,input_size=(3,224,224))
import torch
import torchvision
!pip install torchsummary
import torchsummary
net = torchvision.models.vgg19()
查看模型结构
print(net)
参数量、各层输入输出矩阵大小
torchsummary.summary(model=net,input_size=(3,224,224))