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
4755400eda
refactor(mge/quantization): add `narrow_range` to control quant dtype's lower bound
GitOrigin-RevId: 92389341be
5 years ago
Megvii Engine Team
fda9599a84
feat(mge/quant): add TQT quant method
GitOrigin-RevId: 00b1616e73
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
e6820b91e8
feat(mge/module): add conv and conv_relu quantization module
GitOrigin-RevId: 9cd668d97b
5 years ago
Megvii Engine Team
40412e2688
feat(mge/module): add sync bn
GitOrigin-RevId: ae71a540d1
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
bb872965e3
feat(mge/module): add local conv2d module
GitOrigin-RevId: bb5ab94b19
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
855c49ca7d
feat(mge/module): add linear quantization module
GitOrigin-RevId: d0c96a9411
5 years ago
Megvii Engine Team
980ebf2c72
feat(mge/module): add fused conv_bn qat approximate version
GitOrigin-RevId: 1b7284a595
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
ae506ecfce
docs(mge/module): fix momentum equation in batch norm doc
GitOrigin-RevId: 747ae1646a
5 years ago
Megvii Engine Team
b2ed2630f7
Revert "fix(mge/functional): rm useless arguments"
This reverts commit 8e4f25bfd8
.
GitOrigin-RevId: df696ab8a2
5 years ago
Megvii Engine Team
d23fec06cb
fix(mge/functional): rm useless arguments
group and compute_mode are not used by local_conv2d
GitOrigin-RevId: 8e4f25bfd8
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
4997270156
docs(mge/module): refine the docstring of several apis
GitOrigin-RevId: ea04e05be4
5 years ago
Megvii Engine Team
dd962a4700
fix(mge/module): add negative slope attribute for LeakyReLU module
GitOrigin-RevId: 67e1967b56
5 years ago
Megvii Engine Team
99fae95e02
feat(mge/parampack): add user-defined key to pack params
GitOrigin-RevId: 7d51dcae23
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
779ed45d74
docs(mge/parampack): refine parampack doc
GitOrigin-RevId: 1cff83c308
5 years ago
Megvii Engine Team
7ed98dd50a
feat(mge/functional): add conv_transpose2d support
GitOrigin-RevId: 9ad87a4ea9
5 years ago
Megvii Engine Team
f077a5292c
fix(mge/param_pack): release old parameters
GitOrigin-RevId: 40a1f044e9
5 years ago
Megvii Engine Team
48822557b8
docs(mge/bn): fix docs and tests of batchnorm
GitOrigin-RevId: 8a96aa5fc2
5 years ago
Megvii Engine Team
2c4d1afe14
test(mge): move `NUMBER` config to pytest.init
GitOrigin-RevId: 1d82209c40
5 years ago
Megvii Engine Team
e1b2d31d36
test(mge): skip some doctests having different results on gpu
GitOrigin-RevId: 66e5db5c89
5 years ago
Megvii Engine Team
62b0c6cd27
style(mge): apply format.sh
GitOrigin-RevId: a900b1bb6e
5 years ago
Megvii Engine Team
ea00b57d93
refactor(mge/module): use mge rng to sample from uniform/normal distribution
GitOrigin-RevId: 6ec8f99af5
5 years ago
Megvii Engine Team
a1597cfc46
docs(mge/param_pack): add description of param pack and add param pack in example run_dist
GitOrigin-RevId: 1c900e159d
5 years ago
Megvii Engine Team
80d3a06f9e
fix(mge/param_pack): fix param pack with just one param
GitOrigin-RevId: 0d28a12e59
5 years ago
Megvii Engine Team
a744b3cb8a
feat(mge/module):add param pack
GitOrigin-RevId: 9cf1dbe44d
5 years ago
Megvii Engine Team
65432d3b93
fix(mge/module): fix torch subgraph under jit.trace with symbolic=False
GitOrigin-RevId: a208ba79d9
5 years ago
Megvii Engine Team
f91881ffdc
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
5 years ago