pytorch 在Python中从torch获取所有层,包括操作符

gstyhher  于 2023-04-06  发布在  Python
关注(0)|答案(1)|浏览(140)

我想得到所有的层,如卷积等...和操作符,如“添加”在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

时的所有内容

8yoxcaq7

8yoxcaq71#

named_parameters替换为named_modules并打印module(在您的示例中为param)而不是param.shape

相关问题