查看中间变量
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))
示意图