Megvii Engine Team
|
bb6030ff54
|
chore(release): bump version
|
4 years ago |
Megvii Engine Team
|
b689b4ce5c
|
fix(mge/functional): add F.clamp to __all__ to fix docs
GitOrigin-RevId: 8857e2316b
|
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
|
b2f05bf8d8
|
fix(mge/module): fix quantized fold weight value range limit for fused conv/bn modules
GitOrigin-RevId: 007c2f13b6
|
4 years ago |
Megvii Engine Team
|
0df74604bd
|
fix(data): fix contiguous id
GitOrigin-RevId: 7f79cda0b5
|
4 years ago |
Megvii Engine Team
|
c8f3742f0e
|
fix(mge/doc): fix wrong doc for megengine doc
GitOrigin-RevId: a38dc8e102
|
5 years ago |
Megvii Engine Team
|
f856b17085
|
fix(mge/functional): reshape bias to (1, out_features) in linear
GitOrigin-RevId: a15880b7fc
|
5 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
|
2b3a88d011
|
feat(mge/distributed): add parameter replica_mode
GitOrigin-RevId: 244e4ca437
|
5 years ago |
Megvii Engine Team
|
6972fc7d75
|
feat(opr/standalone): import NMSKeep from MegSkull contrib
1. clang-format.
2. Use name "NMSKeepMGB" during serialization to avoid conflict with
MegSkull contrib one.
GitOrigin-RevId: 627c60e482
|
5 years ago |
Megvii Engine Team
|
5fa444c39a
|
refactor(mge/quantization): add `narrow_range` to control quant dtype's lower bound
GitOrigin-RevId: 92389341be
|
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 |
Xu Xinran
|
b532284cc2
|
chore(version): bump version.
|
5 years ago |
Megvii Engine Team
|
af3de7e136
|
fix(jit): fix jit doc and add NCHW44_DOT
GitOrigin-RevId: 5f5feae8e7
|
5 years ago |
Megvii Engine Team
|
7b8a99517c
|
feat(mge/utils): export _internal.plugin to mge.utils
GitOrigin-RevId: 795821fe7f
|
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
|
cc4e1dfd7c
|
feat(mgb/compnode): expose mem_status and try_coalesce_all_free_memory to python
GitOrigin-RevId: 0ec2d556b1
|
5 years ago |
Megvii Engine Team
|
7aeb4f6ca7
|
fix(mge/optimizer): use static key to avoid mem leak
GitOrigin-RevId: 85298084a3
|
5 years ago |
Megvii Engine Team
|
ca52a93e9f
|
fix(mge/quant): fix init value of histogram observer
GitOrigin-RevId: 9c8caa5f8f
|
5 years ago |
Megvii Engine Team
|
38f7cbd9aa
|
fix(mge/module): fix redundant recursion in `train()`
GitOrigin-RevId: 6b3566930b
|
5 years ago |
Megvii Engine Team
|
5c2323529d
|
test(mge/quantization): add `quantize_disabled` related test
GitOrigin-RevId: f62ba600c5
|
5 years ago |
Megvii Engine Team
|
ab91302515
|
feat(mge/quantization): add `quantize_disabled` attribute in Module
GitOrigin-RevId: f108f03c5a
|
5 years ago |
Megvii Engine Team
|
08ac685e3a
|
feat(mge/functional): add logsumexp
GitOrigin-RevId: e7ef50e9ec
|
5 years ago |
Megvii Engine Team
|
01092feb9b
|
feat(mgb): add PackAllReducePass
GitOrigin-RevId: 59c1b45393
|
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
|
b8d8886e35
|
refactor(mge/tensor): combine Dict and TensorDict
GitOrigin-RevId: 6b6c03c04b
|
5 years ago |
Megvii Engine Team
|
7751a0676e
|
docs(mge/tensor): add advanced index related docs
GitOrigin-RevId: 31735ddac4
|
5 years ago |
Megvii Engine Team
|
6742a58b7e
|
fix(quant): observer do not use cond_take
GitOrigin-RevId: a954814bcb
|
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
|
62753c4d30
|
fix(mge/sdk): fix comp_node bug in dump_with_testcast_mge
GitOrigin-RevId: 26a8dc50b8
|
5 years ago |
Megvii Engine Team
|
adfa468899
|
fix(mge/functional): fix scatter doctest failed for GPU platform issue
GitOrigin-RevId: b5f92c39dd
|
5 years ago |
Megvii Engine Team
|
84068a6bb1
|
fix(mge/data): fix typos in voc and objects365
GitOrigin-RevId: 491e607b7e
|
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
|
30b3d3aa3e
|
fix(dnn/gopt): add convolution nchw44-dot format gopt
GitOrigin-RevId: e8e1e96379
|
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
|
15d1bd4d1f
|
fix(mge/quantization): use list rather than generator to modify submodules
GitOrigin-RevId: b5a3851330
|
5 years ago |
Megvii Engine Team
|
7b2c5a7383
|
fix(mge/quantization): fix histogram observer load and store issue
GitOrigin-RevId: b0a2b476e4
|
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
|
ff341cb19b
|
fix(mge/quantization): modify observer api
GitOrigin-RevId: 7b9c22be96
|
5 years ago |
Megvii Engine Team
|
efb60be289
|
fix(mge/distributed): fix get_free_ports return value
GitOrigin-RevId: 8a546855c4
|
5 years ago |
Megvii Engine Team
|
1d7fcecab2
|
feat(mge/serialization): add map location
GitOrigin-RevId: 4b6d83365b
|
5 years ago |
Megvii Engine Team
|
bb872965e3
|
feat(mge/module): add local conv2d module
GitOrigin-RevId: bb5ab94b19
|
5 years ago |
Megvii Engine Team
|
eee3e5594c
|
feat(mge/distributed): add multiprocess launcher
GitOrigin-RevId: 7d831d125f
|
5 years ago |
Megvii Engine Team
|
35d46dbb41
|
fix(mge/functional): simplify the api of add/remove_axis
GitOrigin-RevId: 2482529704
|
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 |