提交 1daeba76 编写于 作者: M Megvii Engine Team

fix(mge/traced_module): fix bn opdef loader

GitOrigin-RevId: d73f953ef6c19ade49e50883e63cefbd7f66ea80
上级 2d20f937
......@@ -130,7 +130,8 @@ def convbn2d_module_loader(expr):
@register_opdef_loader(BatchNorm)
def bn_opdef_loader(expr):
# mge 1.6
if not hasattr(expr, "version"):
if not hasattr(expr, "version") and len(expr.outputs) != 6:
assert len(expr.outputs) == 5
output = expr.outputs[-1]
oup = TensorNode(expr, shape=(0,), dtype=None, qparams=output._qparams,)
expr.outputs.insert(4, oup)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册