我想得到所有的层,比如卷积等......以及像 python 中的“add”这样的运算符,但我真的不知道怎么做。
这是我的代码
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)
它工作正常,但问题是我只能看到像卷积这样的层,而不是所有的运算符。我该如何解决?
将
named_parameters
替换为named_modules
并打印module
(在您的情况下为param
)而不是param.shape