Hua 90d6423551 Replace AdaptiveAvgPool2d with fixed-kernel AvgPool2d for ONNX compatibility ...

AdaptiveAvgPool2d with None dimensions can cause issues with some ONNX
runtimes. Use AvgPool2d with kernel=(img_h//16, 1) to achieve the same
height-to-1 reduction with full ONNX compatibility.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

2026-03-11 13:58:41 +08:00

5.2 KiB

Raw Blame History

View Raw