Fix PSPNet ONNX export issue#1270

Merged

qubvel merged 4 commits intoqubvel-org:mainfrom

lapertor:fix/pspnet-onnx-export

Mar 19, 2026

Commits on Mar 9, 2026

fix(pspnet): fix ONNX export by replacing AdaptiveAvgPool2d with interpolate
lapertor
committed

Commits on Mar 11, 2026

fix: restore state_dict compatibility and handle ONNX export in forward
lapertor
committed

Commits on Mar 12, 2026

Update segmentation_models_pytorch/decoders/pspnet/decoder.py
qubvel
authored

Commits on Mar 13, 2026

fix: support ONNX export and TorchScript for PSPNet
lapertor
committed

⚡