提交 a7e5f64e 编写于 作者: 刘托

Merge branch 'fix-fold_deconv_bn' into 'master'

fix fold deconv and bn

See merge request !969
...@@ -628,14 +628,13 @@ class Transformer(base_converter.ConverterInterface): ...@@ -628,14 +628,13 @@ class Transformer(base_converter.ConverterInterface):
framework = ConverterUtil.get_arg( framework = ConverterUtil.get_arg(
op, MaceKeyword.mace_framework_type_str).i op, MaceKeyword.mace_framework_type_str).i
input_len = len(op.input) input_len = len(op.input)
if consumer_op.type == MaceOp.BatchNorm.name and \ if consumer_op.type == MaceOp.BatchNorm.name and (
(framework == FrameworkType.CAFFE.value and (framework == FrameworkType.CAFFE.value and
(input_len == 2 or (input_len == 2 or (input_len == 3 and
(input_len == 3 and op.input[-1] in self._consts))) or
op.input[-1] in self._consts))) or \
(framework == FrameworkType.TENSORFLOW.value and (framework == FrameworkType.TENSORFLOW.value and
(input_len == 3 or (input_len == 4 and (input_len == 3 or (input_len == 4 and
op.input[-1] in self._consts))): op.input[-1] in self._consts)))):
print("Fold deconv and bn: %s(%s)" % (op.name, op.type)) print("Fold deconv and bn: %s(%s)" % (op.name, op.type))
filter = self._consts[op.input[1]] filter = self._consts[op.input[1]]
scale = self._consts[consumer_op.input[1]] scale = self._consts[consumer_op.input[1]]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册