是用自己的CUDA训练的pth文件，转换为onnx，再转为om文件后，想在板子上推理。使用样例给的mindx.sdk的base进行推理，但是infer的过程中报错，错误为list out of range。这是源代码。

这是报错误

这是模型的netron里的模型图。

我想请教一下，自己的pc运行的cuda模型推理得到的pth权重，可不可以直接在转换为om模型后在板子上进行推理？即PC上训练，板子上推理。