search

AmigoCDT / MXNet-MobileNetV3

A Gluon implement of MobileNetV3
Apache License 2.0
28 stars 10 forks source link

Convert to ONNX #4

Open wangxiaofanw opened 4 years ago

wangxiaofanw commented 4 years ago

Can this trained model convert to onnx ? when i convert , i face the problem: No conversion function registered for op type _contrib_AdaptiveAvgPooling2D yet.

JokaHD commented 4 years ago

How about using mxnet.gluon.nn.GlobalAvgPool2D while loading model for converting to onnx?