onnx2torch
onnx2torch copied to clipboard
Does not support non symmetric padding techniques.
When converting a model from onnx to pytorch I get the error,
~/.local/lib/python3.8/site-packages/onnx2torch/converter.py in convert(onnx_model_or_path, save_input_names, attach_onnx_mapping)
120 )
121
--> 122 torch_module, onnx_mapping = converter(onnx_node, onnx_graph)
123 if attach_onnx_mapping:
124 setattr(torch_module, 'onnx_mapping', onnx_mapping)
~/.local/lib/python3.8/site-packages/onnx2torch/node_converters/conv.py in _(node, graph)
48 dilation=node_attributes.get('dilations', 1),
49 groups=node_attributes.get('group', 1),
---> 50 padding=onnx_padding_to_torch_padding(
51 node_attributes.get('pads', [0] * spatial_rank * 2),
52 node_attributes.get('auto_pad', 'NOTSET'),
~/.local/lib/python3.8/site-packages/onnx2torch/utils/common.py in onnx_padding_to_torch_padding(padding, auto_pad)
80 half_len = len(padding) // 2
81 if padding[:half_len] != padding[half_len:]:
---> 82 raise NotImplementedError(f'Only symmetric padding is implemented ({padding})')
83
84 padding = padding[:half_len]
NotImplementedError: Only symmetric padding is implemented ([0, 0, 1, 1])```
Seems a similar topic was discussed in https://github.com/ENOT-AutoDL/onnx2torch/issues/52.