PTL to ONNX export() got an unexpected keyword argument 'example_outputs'

[davodogster]

Bug description

Im using torch version 1.12 and received this error when trying to convert this SMP Timm DeepLab model:

How to reproduce:

Do you know how to fix this? Thanks, Sam

How to reproduce the bug

No response

Error messages and logs

# Error messages and logs here please

Environment

More info

No response

[rohitgr7]

this issue got fixed in v1.5.7: #11116, mind try that?
Or we always recommend using the latest release :)

[davodogster]

Thanks that seemed to work

[davodogster]

@rohitgr7 Hi Rohit, the Hitman, Gupta and @awaelchli Do you know how I can successfully load and use the model? It was trained in segmentation_models.pytorch with lightning (v 1.5.4)
This code fails:

Thanks, Sam

[rohitgr7]

you might have to provide a sample input for these models.
Check this: qubvel-org/segmentation_models.pytorch#272 (comment)

[davodogster]

Hi @rohitgr7 thanks for the reply! I figured it out recently and was able to load the model and use it for inference like this:

I was surprised that it worked but the predictions seem legitimate

[rohitgr7]

cool! feel free to close this issue if you are good on your side :)