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
|
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
|
bb872965e3
|
feat(mge/module): add local conv2d module
GitOrigin-RevId: bb5ab94b19
|
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
|
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
|
814a427810
|
feat(mge/module): add `with_parent` argument in `_flatten`
GitOrigin-RevId: 1c88559ece
|
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
|
712b87c8c1
|
feat(mgb/core): add comp node for cambricon
add testcase for cambricon comp node
GitOrigin-RevId: 7794faa47f
|
5 years ago |
Megvii Engine Team
|
e8494f729b
|
fix(unittest): remove the torch dependency in test_conv
GitOrigin-RevId: 36dcab2b95
|
5 years ago |
Megvii Engine Team
|
7ed98dd50a
|
feat(mge/functional): add conv_transpose2d support
GitOrigin-RevId: 9ad87a4ea9
|
5 years ago |
Megvii Engine Team
|
dd41e2e47a
|
fix(mge/test):fix integration/manual tests
GitOrigin-RevId: 5933392407
|
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 |