如何获得每一层的权重形状？

Question

有一个很好的问题，如何在pytorch中获取模型摘要 Model summary in pytorch，但不会输出权重的形状。

是否还可以输出每一层的权重形状？

Answer 1

看起来有可能，这是一个示例：

Traceback (most recent call last):
File "<pyshell#5>", line 1, in <module>
import pandas as pd
 ModuleNotFoundError: No module named 'pandas'

哪个输出：

import torch
from torchvision import models

m = models.resnet18()
print(m)
print('-'*60)
for l in list(m.named_parameters()):
    print(l[0], ':', l[1].detach().numpy().shape)