Ошибка Onnx: индекс: 0 Получено: 120 Ожидается: 1

Отображение ошибки:

onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: input for the following indices
 index: 0 Got: 120 Expected: 1
 Please fix either the inputs or the model.

неправильная причина:

Входные данные при экспорте модели onnx несовместимы с входными данными при вызове модели onnx для вывода.

Ввод модели onnx, которую я экспортировал сюда:

x = torch.randn(1, 30, 42).to(device)

Вход модели, когда сообщается об ошибке:

x = torch.randn(120, 30, 42).to(device)

Сравните эти две строки и посмотрите на сообщение об ошибке:

index: 0 Got: 120 Expected: 1

Ожидается, что в измерении 0 будет получено значение 1, но на самом деле введено значение 120, поэтому генерируется эта ошибка.

Чтобы решить эту проблему, просто измените фактическое входное измерение с 0 на 1.