Megvii Engine Team
3f2770fe03
feat(mge/module): add quantize dtype load support for module load_state_dict
GitOrigin-RevId: 0a94cb6b17
5 years ago
Megvii Engine Team
38f7cbd9aa
fix(mge/module): fix redundant recursion in `train()`
GitOrigin-RevId: 6b3566930b
5 years ago
Megvii Engine Team
5c2323529d
test(mge/quantization): add `quantize_disabled` related test
GitOrigin-RevId: f62ba600c5
5 years ago
Megvii Engine Team
ab91302515
feat(mge/quantization): add `quantize_disabled` attribute in Module
GitOrigin-RevId: f108f03c5a
5 years ago
Megvii Engine Team
d2f5874a52
fix(mge/module): fix non-str key error of dict in module
GitOrigin-RevId: f82cd48230
5 years ago
Megvii Engine Team
caf1fac251
refactor(mge/quantization): split `QATModule` and refactor convert api
GitOrigin-RevId: 80cfb12d10
5 years ago
Megvii Engine Team
ff341cb19b
fix(mge/quantization): modify observer api
GitOrigin-RevId: 7b9c22be96
5 years ago
Megvii Engine Team
205291a3b5
feat(mge/quantization): add histgram observer
GitOrigin-RevId: a9252a6baf
5 years ago
Megvii Engine Team
7c4f1a3851
feat(mge/quantization): add calibration support
GitOrigin-RevId: f16fbba2b7
5 years ago
Megvii Engine Team
38a95744fb
refactor(mge/quantization): refactor qconfig, remove inp_observer and bias_fakequant
GitOrigin-RevId: e57f9edd12
5 years ago
Megvii Engine Team
de75dae84c
fix(mge/quantization): fix get scale issue
GitOrigin-RevId: 99068d7422
5 years ago
Megvii Engine Team
206521cd33
fix(mge/quant): avoid updating the scale and zero point in eval mode
GitOrigin-RevId: dfb08cf701
5 years ago
Megvii Engine Team
8c110c3942
feat(mge/quantization): add quantization interface
GitOrigin-RevId: 4fe1233ec3
5 years ago
Megvii Engine Team
814a427810
feat(mge/module): add `with_parent` argument in `_flatten`
GitOrigin-RevId: 1c88559ece
5 years ago
Megvii Engine Team
9a42c63641
fix(mge/module): add the more warnings in load_state_dict
In the before, the information of miss matching and unused operators won't be printed in the non-trict mode, this commit add the information.
GitOrigin-RevId: b2543eb832
5 years ago
Megvii Engine Team
62b0c6cd27
style(mge): apply format.sh
GitOrigin-RevId: a900b1bb6e
5 years ago
Megvii Engine Team
a744b3cb8a
feat(mge/module):add param pack
GitOrigin-RevId: 9cf1dbe44d
5 years ago
Megvii Engine Team
f91881ffdc
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
5 years ago