From 414a543241f92d90465d204afb1923e196ebb5cb Mon Sep 17 00:00:00 2001 From: Macrobull Date: Fri, 29 Mar 2019 13:27:09 +0800 Subject: [PATCH] fix required ksize=-1 in global pool --- onnx2fluid/onnx2fluid/symbolic.py | 1 + 1 file changed, 1 insertion(+) diff --git a/onnx2fluid/onnx2fluid/symbolic.py b/onnx2fluid/onnx2fluid/symbolic.py index b45794a..fa40738 100644 --- a/onnx2fluid/onnx2fluid/symbolic.py +++ b/onnx2fluid/onnx2fluid/symbolic.py @@ -413,6 +413,7 @@ def _global_pool(prog, pool_type, inputs, outputs, attrs, value_infos, name=''): global_pooling=True, adaptive=False, pooling_type=pool_type, + ksize=[-1, -1], ), ) -- GitLab