使用ONNXRuntime加载ONNX模型时出错

Question

我正在将一个自定义的Pytorch模型转换为ONNX。但是，当使用ONNXRuntime加载它时，我遇到了如下错误：

onnxruntime.capi.onnxruntime_pybind11_state.RuntimeException: [ONNXRuntimeError] : 6 : RUNTIME_EXCEPTION : Exception during initialization: ...onnxruntime/core/providers/cpu/tensor/transpose.h:46 onnxruntime::TransposeBase::TransposeBase(const onnxruntime::OpKernelInfo &) v >= 0 && static_cast<uint64_t>(v) <= std::numeric_limits<size_t>::max() was false.

我已经和onnx.checker.check_model()确认过了，完全没问题。我还尝试在forward()函数中将transpose()替换为permute()，但错误仍然存在。

有人熟悉这个错误吗？

环境：

Python 3.7
Pytorch 1.9.0
CUDA 10.2 
ONNX 1.10.1
ONNXRuntime 1.8.1 
OS Ubuntu 18.04

Answer 1

节点Transpose_52的perm属性为-1，0，1，尽管ONNX Runtime要求所有这些属性都应为正：onnxruntime/core/providers/cpu/tensor/transpose.h#L46