From 1daeba76fdddd5c23377aca5653947ec2eff5c88 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Thu, 25 Nov 2021 10:47:02 +0800 Subject: [PATCH] fix(mge/traced_module): fix bn opdef loader GitOrigin-RevId: d73f953ef6c19ade49e50883e63cefbd7f66ea80 --- imperative/python/megengine/traced_module/compat.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/imperative/python/megengine/traced_module/compat.py b/imperative/python/megengine/traced_module/compat.py index 878d2fc6..acd4afa0 100644 --- a/imperative/python/megengine/traced_module/compat.py +++ b/imperative/python/megengine/traced_module/compat.py @@ -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)