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
|
c8a9094bb3
|
fix(mge/data/dataloader): add refcount in _PlasmaStoreManager
GitOrigin-RevId: 9a95cb1a5d
|
5 years ago |
Megvii Engine Team
|
fda9599a84
|
feat(mge/quant): add TQT quant method
GitOrigin-RevId: 00b1616e73
|
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
|
6d0d5e5a75
|
fix(mge/redis_server): fix bug for external connection
GitOrigin-RevId: 0b595712d7
|
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
|
dd8f3ffc4d
|
test(distributed): replace python command in subprocess with `sys.executable`
GitOrigin-RevId: 1f4262fd0c
|
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
|
cec7fd3c5f
|
fix(mgb/build): fix multi-machine macro and add test_distributed
GitOrigin-RevId: cb1bfe8742
|
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
|
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
|
0bdb64c5f9
|
ci(test): dump test time result to json file
GitOrigin-RevId: 0ed092a0e8
|
5 years ago |
Megvii Engine Team
|
888895e971
|
refactor(mge/test): make test_parampack more stable
GitOrigin-RevId: d82230ea07
|
5 years ago |
Megvii Engine Team
|
fd83e0252d
|
fix(mge/test): fix reproducibility of test_correctness.py
GitOrigin-RevId: 70bd43bbab
|
5 years ago |