分類

展開全部 | 收合全部

分類

展開全部 | 收合全部

【Pytorch】分析 torch model (torchscript) 每一層的內容

內容目錄

前言

分析 torch model 的中間層與 shape

範例

import torch

model_path="model.pt"
model = torch.jit.load(model_path)
print(model.graph)

inputs=list(model.graph.inputs())
outputs=list(model.graph.outputs())

# model quote
print(" ------- quote of the model ------- ")
print(f"{inputs[0]=}")
inputs=inputs[1:]   

print(" ------- input layer ------- ")
for i, input in enumerate(inputs):
    print(f"[input {i}]")
    print(f"{input.debugName()=}")


print(" ------- intermediate layer ------- ")
for name, param in model.named_parameters():
    print(f"Layer: {name}, Shape: {param.shape}")


print(" ------- output layer ------- ")
for i, output in enumerate(outputs):
    print(f"[output {i}]")
    print(f"{output.debugName()=}")
Howard Weng
Howard Weng

我是 Howard Weng,很多人叫我嗡嗡。這個網站放了我的各種筆記。希望這些筆記也能順便幫助到有需要的人們!如果文章有幫助到你的話,歡迎幫我點讚哦!
另外,因為定位是「個人的隨手筆記」,有些文章內容「⚠️可能我理解有誤⚠️」或「?只寫到一半?」,如果有發現這樣的情況,歡迎在該文章的最下面留言提醒我!我會儘快修正或補上!感謝大家的建議與幫忙,讓網站能變得更好?

文章: 866