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 |
Megvii Engine Team
|
20b67b294c
|
ci(python): add format and lint check for python
GitOrigin-RevId: d8d106b927
|
5 years ago |
Megvii Engine Team
|
62b0c6cd27
|
style(mge): apply format.sh
GitOrigin-RevId: a900b1bb6e
|
5 years ago |
Megvii Engine Team
|
c9830d20d0
|
fix(mge/test): fix tolerance
GitOrigin-RevId: 58c029b394
|
5 years ago |
Megvii Engine Team
|
2174fd12b9
|
refactor(mge/test): make test_converge more stable
GitOrigin-RevId: 710e3ede40
|
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 |
Megvii Engine Team
|
9edce7e294
|
ci(python_test): run python test with cuda
GitOrigin-RevId: bc51c2e4d5
|
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 |