我想得到所有的层,如卷积等...和操作符,如“添加”在python中,但我真的不知道如何。
这是我的密码
import torch
# An instance of your model.
model = torch.jit.load('best5.torchscript')
# Print all the layers
for name, param in model.named_parameters():
print(name, param.shape)
它工作正常,但问题是我只能看到卷积层,而不是所有的操作符。我该如何解决这个问题?
例如,在此图像中,我可以看到模型加载到netron
时的所有内容
1条答案
按热度按时间8yoxcaq71#
将
named_parameters
替换为named_modules
并打印module
(在您的示例中为param
)而不是param.shape