提交 0c08cdf7 编写于 作者: 叶剑武

fix conv_to_fc transformer when weight is reused by other conv

上级 2f949efb
......@@ -1013,7 +1013,8 @@ class Transformer(base_converter.ConverterInterface):
zero_padding = False
if height == filter_height and width == filter_width \
and zero_padding:
and zero_padding \
and len(self._consumers[op.input[1]]) == 1:
print("transform global conv to fc %s(%s)"
% (op.name, op.type))
op.type = MaceOp.FullyConnected.name
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册