TypeError：“GraphModule”对象不可下标（Pytorch 中 .onnx ML 模型的访问权重）

Question

我有一个流行的 .onnx ML 天气预报模型，我正在尝试将其转换为 PyTorch 进行微调。我使用以下代码来转换它：

import os
import numpy as np
import onnx
from onnx import numpy_helper
import onnxruntime as ort
from onnx2torch import convert

model_24 = onnx.load('pangu_weather_24.onnx')
tm = convert(model_24) #Convert onnx model to torch

从这里，我想访问“tm”对象中模型的权重，但我似乎无法在网上找到任何相关资源。

尝试用 tm[0] 对其进行下标会显示以下错误：

TypeError: 'GraphModule' object is not subscriptable

并且通过“tm.dict”该对象的字典更加令人困惑（粘贴在图像中）。

在线访问 PyTorch 权重矩阵的常规方法也揭示了相同的错误，即图形模块不可下标

Answer 1

您可以通过

tm.parameters()

正常访问其权重，就像任何

nn.Module()

一样