pytorch - 使用 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()，完全没问题。我也尝试过在函数中替换transpose()，permute()但forward()错误仍然存​​在。

有人熟悉这个错误吗？

环境：

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

Answer 1

node的perm属性Transpose_52是 [-1, 0, 1] 虽然 ONNX Runtime 要求它们都应该是正数：onnxruntime/core/providers/cpu/tensor/transpose.h#L46