提交 520e0618 编写于 作者: 刘琦

Merge branch 'feature_wuch' into 'master'

fix caffe graph convert

See merge request !396
......@@ -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'
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册