From e13fb7b99be35be83ecef7fc14e7b898baf1e6a1 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 25 Aug 2020 14:51:48 +0800
Subject: [PATCH] fix(mge/quantization): fix apply observer bug which get
 qparam before call observer

GitOrigin-RevId: 7fd633368343f1512e18141509bf4414a6d7e6b2
---
 python_module/megengine/module/qat/module.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python_module/megengine/module/qat/module.py b/python_module/megengine/module/qat/module.py
index 7ebae774..8546e4c6 100644
--- a/python_module/megengine/module/qat/module.py
+++ b/python_module/megengine/module/qat/module.py
@@ -72,11 +72,11 @@ class QATModule(Module):
     ):
         # do observer
         if observer is None:
-            q_dict = None
             oup = target
+            q_dict = None
         else:
-            q_dict = observer.get_qparams()
             oup = observer(target)
+            q_dict = observer.get_qparams()
         # do fake quant
         if fake_quant is not None:
             oup = fake_quant(oup, q_dict)