onnx2torch icon indicating copy to clipboard operation
onnx2torch copied to clipboard

Does not support non symmetric padding techniques.

Open sheikhazhanmohammed opened this issue 2 years ago • 1 comments

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])```

sheikhazhanmohammed avatar Jul 01 '22 09:07 sheikhazhanmohammed

Seems a similar topic was discussed in https://github.com/ENOT-AutoDL/onnx2torch/issues/52.

yan12125 avatar Jul 16 '22 07:07 yan12125