Megvii Engine Team
|
00ef677249
|
fix(mgb): remove internal for cambricon and atlas
GitOrigin-RevId: 861e349eb4
|
4 years ago |
Megvii Engine Team
|
13e8f00a37
|
feat(mge/module): add forward hook support
GitOrigin-RevId: c0db58df13
|
4 years ago |
Megvii Engine Team
|
ef239f835f
|
feat(windows/python_whl): make windows HAPPY for build megbrain python package
GitOrigin-RevId: 92b2c07bf9
|
4 years ago |
Megvii Engine Team
|
8269215959
|
fix(mge/module): fix named_children of Sequential
GitOrigin-RevId: d3220fb361
|
4 years ago |
Megvii Engine Team
|
177c268898
|
feat(osx/python_whl): fix python pytest failedon osx
GitOrigin-RevId: 97d2b496bb
|
4 years ago |
Megvii Engine Team
|
caf77d000e
|
feat(mge/module): add quantize dtype load support for module load_state_dict
GitOrigin-RevId: 0a94cb6b17
|
5 years ago |
Megvii Engine Team
|
4ace67ff44
|
test(mge/distributed): check gpu num for multi gpu test
GitOrigin-RevId: 78f4718682
|
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 |