onnx2torch icon indicating copy to clipboard operation
onnx2torch copied to clipboard

Published 20 hours ago • verified publisher icon ENOT-AutoDL

require to support operator 'ArgMax'

Open Lin-Tao-AI opened this issue 1 year ago • 4 comments

Met the following error, hope to support operator 'ArgMax', thanks

Traceback (most recent call last): File "anna_onnx2torch.py", line 38, in torch_model_1 = convert(onnx_model_path) File "/home/lin/anaconda3/envs/anna/lib/python3.8/site-packages/onnx2torch/converter.py", line 104, in convert converter = get_converter( File "/home/lin/anaconda3/envs/anna/lib/python3.8/site-packages/onnx2torch/node_converters/registry.py", line 69, in get_converter raise NotImplementedError(f'Converter is not implemented ({description})') NotImplementedError: Converter is not implemented (OperationDescription(domain='', operation_type='ArgMax', version=12))

Lin-Tao-AI avatar Mar 06 '23 09:03 Lin-Tao-AI

Hey @Lin-Tao-AI ,

I'm already developing this operator. Actually I'm having some problems, but I think I should solve it soon.

I think we will support it soon.

JohnMasoner avatar Mar 06 '23 14:03 JohnMasoner

This issue is stale because it has been open 30 days with no activity. Remove stale label or comment or this will be closed in 10 days.

github-actions[bot] avatar Apr 15 '23 02:04 github-actions[bot]

This issue was closed because it has been stalled for 10 days with no activity.

github-actions[bot] avatar Apr 25 '23 02:04 github-actions[bot]

Is there any progress on this? I'm facing this issue in trying to round trip a torch model => onnx and then back to torch.

onnx2torch == 1.5.7

If it would help, I can share my model file also.

kochhar-mw avatar Jun 22 '23 11:06 kochhar-mw

Implemented: https://github.com/ENOT-AutoDL/onnx2torch/releases/tag/v1.5.15

senysenyseny16 avatar Aug 07 '24 14:08 senysenyseny16