[aidge_onnx] Max_Pooling error
Converting an Onnx model with maxpooling layers raises the following error:
Traceback (most recent call last):
File "inference/inference.py", line 18, in <module>
model = aidge_onnx.load_onnx(data_dir + "model/default_quantization_model.onnx")
File "/home/aph/Projects/aidge_sandbox/aidge_sandbox/lib/python3.8/site-packages/aidge_onnx/onnx_loader.py", line 30, in load_onnx
return _load_onnx2graphview(model, verbose)
File "/home/aph/Projects/aidge_sandbox/aidge_sandbox/lib/python3.8/site-packages/aidge_onnx/onnx_loader.py", line 70, in _load_onnx2graphview
model_nodes[node_name] = ONNX_NODE_CONVERTER_[onnx_node.op_type.lower()](onnx_node, model_inputs[node_name])
File "/home/aph/Projects/aidge_sandbox/aidge_sandbox/lib/python3.8/site-packages/aidge_onnx/node_converter/converters/maxpool.py", line 39, in import_max_pooling
max_pooling_node = aidge_core.__getattribute__(f"MaxPooling{len(kernel_dims)}D")(
TypeError: MaxPooling2D(): incompatible function arguments. The following argument types are supported:
1. (kernel_dims: List[int], name: str = '', stride_dims: List[int] = [1, 1], padding_dims: List[int] = [0, 0, 0, 0]) -> aidge_core.aidge_core.Node
Invoked with: [2, 2]; kwargs: name='model_20/max_pooling2d_52/MaxPool', stride_dims=[2, 2], padding_dims=0
I can provide the model summary, or even the model if it is needed to test the problem.
Edited by Clément Fisher