gpt4 book ai didi

protocol-buffers - 如何从 ONNX 模型中读取各个层的权重和偏差值?

转载 作者:行者123 更新时间:2023-12-05 09:15:27 36 4
gpt4 key购买 nike

如何从 ONNX 模型中获取权重/偏置矩阵值,我目前可以从 model.onnx 中获取输入、内核大小、步幅和填充值。我加载模型,然后读取图形节点以获得相同的结果:

import onnx
m = onnx.load('model.onnx')
print(m.graph.node)

最佳答案

from onnx import numpy_helper
MODEL_PATH = "....../resnet50"
_model = onnx.load(MODEL_PATH + "/model.onnx")
INTIALIZERS=_model.graph.initializer
Weight=[]
for initializer in INTIALIZERS:
W= numpy_helper.to_array(initializer)
Weight.append(W)

关于protocol-buffers - 如何从 ONNX 模型中读取各个层的权重和偏差值?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52402448/

36 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com