Pytorch查看模型中间变量参数

查看中间变量

for parameters in net.parameters():
    print(parameters)

for name,parameters in net.named_parameters():
    print(name,':',parameters)
    print(name, ':', parameters.size())

示意图

2,## 查看每层的size

import torchvision.models as models
from torchsummary import summary
 
 
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
vgg = models.vgg19().to(device)
 
summary(vgg, (3, 224, 224))

示意图