Browse Source

fix(mge/quantization): fix apply observer bug which get qparam before call observer

GitOrigin-RevId: 7fd6333683
tags/v1.0.0-rc1
Megvii Engine Team Xinran Xu 4 years ago
parent
commit
e13fb7b99b
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      python_module/megengine/module/qat/module.py

+ 2
- 2
python_module/megengine/module/qat/module.py View File

@@ -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)


Loading…
Cancel
Save