Onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError]: 2: INVALID_ARGUMENT ⇐ Python

[Anonymous]

Я попытался преобразовать модель факела в onnx и получил следующую ошибку:

Код: Выделить всё

raise errors.UnsupportedOperatorError(
torch.onnx.errors.UnsupportedOperatorError: ONNX export failed on an operator with unrecognized namespace 'torch_scatter::scatter_max'. If you are trying to export a custom operator, make sure you registered it with the right domain and version.

Поэтому я использовал следующий код для регистрации scatter_max

Код: Выделить всё

def scatter_max(
src: torch.Tensor, index: torch.Tensor, dim: int = -1,
out: Optional[torch.Tensor] = None,
dim_size: Optional[int] = None,fill_value=0) -> Tuple[torch.Tensor, torch.Tensor]:
return torch.ops.torch_scatter.scatter_max(src, index, dim, out, dim_size)

# Register custom symbolic function
torch.onnx.register_custom_op_symbolic("torch_scatter::scatter_max", scatter_max,9)

Я получил файл onnx, но когда я запускаю демо-версию, я получаю следующую ошибку:

Код: Выделить всё

onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Non-zero status code returned while running ScatterElements node. Name:'/hsn/ScatterElements_1' Status Message: Indices vs updates dimensions differs at position=1 1 vs 64

Что я делаю не так?


Подробнее здесь: https://stackoverflow.com/questions/757 ... ntimeerror