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
|
ef437f69aa
|
fix(mge/data/dataloader): add refcount in _PlasmaStoreManager
GitOrigin-RevId: 9a95cb1a5d
|
5 years ago |
Megvii Engine Team
|
6d367454cf
|
feat(mge/opr-mm): add param local_grad for collective_comm opr
GitOrigin-RevId: cc120cfb55
|
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
|
6ed791bc86
|
feat(mge/quant): add TQT quant method
GitOrigin-RevId: 00b1616e73
|
5 years ago |
Megvii Engine Team
|
328fb36f17
|
feat(mgb/opr-mm): add Scatter, Gather, AllToAll oprs
GitOrigin-RevId: f75169ecd6
|
5 years ago |
Megvii Engine Team
|
5c2323529d
|
test(mge/quantization): add `quantize_disabled` related test
GitOrigin-RevId: f62ba600c5
|
5 years ago |
Megvii Engine Team
|
08ac685e3a
|
feat(mge/functional): add logsumexp
GitOrigin-RevId: e7ef50e9ec
|
5 years ago |
Megvii Engine Team
|
c7e6c658fd
|
refactor(mge/distribute): use is_root (and rank) in stead of rank and root at collective comm
GitOrigin-RevId: dccdb71553
|
5 years ago |
Megvii Engine Team
|
35bc0e1f60
|
fix(mge/function): do not deeply copy saved tensor in Function
GitOrigin-RevId: 3c89d1ceaa
|
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
|
f7d8b51648
|
feat(mge/functional): add smooth l1 loss
GitOrigin-RevId: c1437788d7
|
5 years ago |
Megvii Engine Team
|
3c49d1d324
|
feat(mge/functional): add hinge loss
GitOrigin-RevId: 64c89c1f8c
|
5 years ago |
Megvii Engine Team
|
7aa7a09b6b
|
feat(mge/optimizer): add optimizer adadelta
GitOrigin-RevId: 244bc0d74a
|
5 years ago |
Megvii Engine Team
|
5eca4da323
|
feat(mge/functional): add softplus function
GitOrigin-RevId: 26c1ab7416
|
5 years ago |
Megvii Engine Team
|
90107b6dda
|
feat(mge/cgtools): add network vistior interface with optional pruning
GitOrigin-RevId: cfa69e3e83
|
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
|
3adad4852d
|
feat(mge/optimizer): add optimizer adagrad
GitOrigin-RevId: 60ff08c5c3
|
5 years ago |
Megvii Engine Team
|
8be78b114e
|
test(mge/optimizer): refactor the unittest of optimizer
GitOrigin-RevId: 4754285713
|
5 years ago |
Megvii Engine Team
|
b3b14fdfe1
|
fix(mge/jit): fix add_update semantic
GitOrigin-RevId: f541ac7c6d
|
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
|
4500faf1ea
|
fix(python_module/mge): one_hot: no default value for num_classes
GitOrigin-RevId: c4a5310880
|
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
|
d3730036ea
|
test(mge/dist): add unit tests for megengine.distributed.utils
GitOrigin-RevId: 32336c66f1
|
5 years ago |
Megvii Engine Team
|
80d9e56b38
|
fix(mge/tensor): where support x or y is empty
GitOrigin-RevId: 2b69ee2e79
|
5 years ago |
Megvii Engine Team
|
4c5d01fa34
|
refactor(mge/examples): add example to use modified sublinear API
GitOrigin-RevId: c6f6c65d0b
|
5 years ago |
Megvii Engine Team
|
396653df5f
|
feat(mge/api): expose sublinear related parameters at mge api level
GitOrigin-RevId: 7a47f5d0d5
|
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
|
5507a29bd3
|
feat(mge/functional): add normalize opr
GitOrigin-RevId: 572a32f563
|
5 years ago |
Megvii Engine Team
|
000663c30c
|
feat(mge/functional): add isnan and isinf oprs
GitOrigin-RevId: b4a347751c
|
5 years ago |
Megvii Engine Team
|
9b41321926
|
fix(mge/functional): support negative axis in math.py
GitOrigin-RevId: 75143a7309
|
5 years ago |
Megvii Engine Team
|
207527d108
|
fix(mge/clamp): fix `F.clamp`
GitOrigin-RevId: 1efac8add6
|
5 years ago |
Megvii Engine Team
|
bb92ee26e3
|
fix(mge/functional): refine doc and add test for linspace
GitOrigin-RevId: d6c7eea2c2
|
5 years ago |
Megvii Engine Team
|
ce30d045ee
|
feat(mge/functional): add arange in tensor.py
GitOrigin-RevId: ad88a4c18e
|
5 years ago |
Megvii Engine Team
|
fc4dedc0bc
|
fix(mge/random): add lower bound and higher bound for uniform sampling
GitOrigin-RevId: 2a2c56fd17
|
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
|
d74a4ee931
|
fix(mge/build): use consistent genfiles/megbrain_build_config.h
Modify get_dynamic_info to work with cmake changes
Also rename get_dynamic_info as get_mem_allocation_info
GitOrigin-RevId: 383affe388
|
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
|
044695aa2a
|
feat(mgb/plugin): add profiler for dynamic graph
GitOrigin-RevId: 10a6cccff9
|
5 years ago |
Megvii Engine Team
|
62b0c6cd27
|
style(mge): apply format.sh
GitOrigin-RevId: a900b1bb6e
|
5 years ago |
Megvii Engine Team
|
dd41e2e47a
|
fix(mge/test):fix integration/manual tests
GitOrigin-RevId: 5933392407
|
5 years ago |
Megvii Engine Team
|
8d2bbf7383
|
fix(mge/functional): support scalar inputs in elemwise functions
GitOrigin-RevId: 7bce561ee1
|
5 years ago |