xception 모델을 onnx로 변환 후 로드하여 양자화하는 과정 중에서 오류가 납니다.
model = onnx.load_model("model.onnx")
optimized_onnx_model = optimize_model(model=model, opset_version=13, input_shapes={"x": [1, 224, 224, 3]},
)
해당 코드에서
Exception has occurred: InvalidArgument
[ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Failed to load model with error: /onnxruntime_src/onnxruntime/core/graph/model.cc:147 onnxruntime::Model::Model(onnx::ModelProto&&, const PathString&, const IOnnxRuntimeOpSchemaRegistryList*, const onnxruntime::logging::Logger&, const onnxruntime::ModelOptions&) Unsupported model IR version: 9, max supported IR version: 8
File “/home/ubuntu/miniconda3/envs/new_env/furiosa/furiosa_quantizer.py”, line 17, in main
optimized_onnx_model = optimize_model(
File “/home/ubuntu/miniconda3/envs/new_env/furiosa/furiosa_quantizer.py”, line 46, in
main()
onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Failed to load model with error: /onnxruntime_src/onnxruntime/core/graph/model.cc:147 onnxruntime::Model::Model(onnx::ModelProto&&, const PathString&, const IOnnxRuntimeOpSchemaRegistryList*, const onnxruntime::logging::Logger&, const onnxruntime::ModelOptions&) Unsupported model IR version: 9, max supported IR version: 8
위와 같이 IR version이 맞지 않는다는 오류가 납니다.
어떻게 해결할 수 있을까요?
감사합니다.