diff --git a/mace/python/tools/caffe_converter_lib.py b/mace/python/tools/caffe_converter_lib.py index 4086cc878faf88d382ea9c626c981be1eb9d333d..cce64c4dce96a956fc89a65e69222656d7159318 100644 --- a/mace/python/tools/caffe_converter_lib.py +++ b/mace/python/tools/caffe_converter_lib.py @@ -800,7 +800,9 @@ class CaffeConverter(object): input_shape = op.get_single_parent().output_shape_map[op.layer.bottom[ 0]] if param.HasField('global_pooling') and param.global_pooling: - kernels = [input_shape[2], input_shape[3]] + kernels = [input_shape[2], input_shape[3]] \ + if self.device == 'neon' else \ + [input_shape[1], input_shape[2]] kernel_arg = op_def.arg.add() kernel_arg.name = 'kernels'