mmdeploy icon indicating copy to clipboard operation
mmdeploy copied to clipboard

Published 20 hours ago • verified publisher icon open-mmlab

fix adaptive_avg_pool exporting to onnx

Open RunningLeon opened this issue 3 years ago • 0 comments

Motivation

add global rewriting for adaptive_avg_pool and ease the exporting to onnx for models such as pspnet, litehrnet.

Modification

  • Add adaptive_avg_pool2d__default rewriting in mmdeploy/pytorch/functions/adaptive_pool.py
  • Remove some symbolic functions for adaptive_avg_pool that are modified from pytorch.
  • Update tests

BC-breaking (Optional)

None

Use cases (Optional)

None

RunningLeon avatar Aug 03 '22 07:08 RunningLeon