segmentation_models.pytorch icon indicating copy to clipboard operation
segmentation_models.pytorch copied to clipboard

Published 20 hours ago • verified publisher icon qubvel-org

Fix code to be able to correctly export MobileOne when using SE layer

Open Marcelo5444 opened this issue 1 year ago • 0 comments

With current code, ONNX export fails miserably due to list creation.

This work around allows to correclty trace torch model and export to ONNX

Marcelo5444 avatar Feb 05 '24 19:02 Marcelo5444