Megvii Engine Team
|
46d96478a1
|
feat(mge/tools): optimize statistical tools
BREAKING CHANGE:
GitOrigin-RevId: cd2a1acd11
|
4 years ago |
Megvii Engine Team
|
7f9d441cbb
|
docs(mge/module): add note about weight shape in conv
GitOrigin-RevId: 43e1f15968
|
4 years ago |
Megvii Engine Team
|
8be70d170c
|
docs(misc): correct docstring format broadly
GitOrigin-RevId: 45234ca07e
|
4 years ago |
Megvii Engine Team
|
53446bcea5
|
fix(imperative/dataset): fix VOC dataset
GitOrigin-RevId: a46a4c1173
|
4 years ago |
Megvii Engine Team
|
8e5bf94857
|
fix(mge/utils): fix bug of VarNode inplace operations
GitOrigin-RevId: fa9eec7079
|
4 years ago |
Megvii Engine Team
|
13b15fb08c
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
6d376623de
|
fix(imperative/dataloader): collect garbage explicitly before starting a process in dataloader
GitOrigin-RevId: 3c3b51ad9c
|
4 years ago |
Megvii Engine Team
|
259894250b
|
feat(imperative/autodiff): add grad clip
GitOrigin-RevId: f344f4a233
|
4 years ago |
Megvii Engine Team
|
601a33a84d
|
refactor(mge/dtr): update dtr api
GitOrigin-RevId: dc366c65be
|
4 years ago |
Megvii Engine Team
|
9b69a02fb2
|
fix(imperative/tensor): init m_offset when constructing a Tensor with DeviceTensorND
GitOrigin-RevId: b340e27c47
|
4 years ago |
Megvii Engine Team
|
226f24fa8e
|
test(mge/distributed): add gather scatter reduce broadcast grad test
GitOrigin-RevId: 8245e11f1d
|
4 years ago |
Megvii Engine Team
|
aea5ac1363
|
fix(mge/distributed): fix gather scatter reduce broadcast autodiff
GitOrigin-RevId: 1c2250a079
|
4 years ago |
Megvii Engine Team
|
b1bf193e4d
|
feat(functional/loss): add reduction choices to loss functions
GitOrigin-RevId: a29e6bb4cf
|
4 years ago |
Megvii Engine Team
|
dadd508615
|
test(param_pack): more cases for param pack concat
GitOrigin-RevId: 0700b548ab
|
4 years ago |
Megvii Engine Team
|
579a3d7786
|
fix(mge/jit): error out if dump bn in training mode
GitOrigin-RevId: edc7ea2962
|
4 years ago |
Megvii Engine Team
|
68da973eec
|
fix(mge/network_visualize): fix warning 'span dist too large'
GitOrigin-RevId: 8db691b8ac
|
4 years ago |
Megvii Engine Team
|
6bb6787d9a
|
feat(mge): add a tool which can analyze the file generated by compare_binary_iodump.py
GitOrigin-RevId: 9acab0a49f
|
4 years ago |
Megvii Engine Team
|
717b88e684
|
fix(mge/elemwise): fix problem that elemwise.mode is not comparable with string mode
GitOrigin-RevId: 82e39be0a9
|
4 years ago |
Megvii Engine Team
|
05550bc5b1
|
fix(mge/quantization): fix a compute problem
GitOrigin-RevId: 8aaedea959
|
4 years ago |
Megvii Engine Team
|
813dd2f1c0
|
fix(mge/core): support non-Tensor args for `core.tensor.utils.get_device`
GitOrigin-RevId: 51dc03db9e
|
4 years ago |
Megvii Engine Team
|
9f5bb42d78
|
perf(fastrun): cache persistent cache
GitOrigin-RevId: 21f7d4c19d
|
4 years ago |
Megvii Engine Team
|
7af1ec66db
|
fix(tensor): del valid tensors when compnode finalizing
GitOrigin-RevId: bace1f2b51
|
4 years ago |
Megvii Engine Team
|
696d2c2ef1
|
fix(mge/autodiff): check tensors to be attached
GitOrigin-RevId: c4f3f80876
|
4 years ago |
Megvii Engine Team
|
984d85ca26
|
feat(mge/functional): argmin and argmax support negtive axis
GitOrigin-RevId: a1bd1102a6
|
4 years ago |
Megvii Engine Team
|
87d6ff228d
|
fix(mge): fix dumping backward graph
GitOrigin-RevId: 430f110053
|
4 years ago |
Megvii Engine Team
|
f31752d5cc
|
feat(mge/module): add __repr__ method for qat and quantized module
GitOrigin-RevId: 0d78cbab9a
|
4 years ago |
Megvii Engine Team
|
9451a96147
|
test(mge/optimizer): update optimizer test to make sure grad not change
GitOrigin-RevId: e207672116
|
4 years ago |
Megvii Engine Team
|
92e2ed6e2e
|
fix(mge/optimizer): fix optimizer update step inplace add grad
GitOrigin-RevId: d677d1ca6b
|
4 years ago |
Megvii Engine Team
|
6972896933
|
fix(mge/utils): fix network multiple outputs issue
GitOrigin-RevId: d22e639cd3
|
4 years ago |
Megvii Engine Team
|
13e6ea349d
|
feat(imperative/opr): rebase rng refactoring to dev & add python module
GitOrigin-RevId: ee5984c52d
|
4 years ago |
Megvii Engine Team
|
cded8ef1a6
|
feat(imperative): add rng opdef
GitOrigin-RevId: b62dcea7f5
|
4 years ago |
Megvii Engine Team
|
c3a1ac3dcf
|
feat(mge/tools): module_status-add-functions
BREAKING CHANGE:
GitOrigin-RevId: ced3da3a12
|
4 years ago |
Megvii Engine Team
|
05e4c8266d
|
fix(mge/module): fix a mistaken in BN docs
GitOrigin-RevId: 086a70649e
|
4 years ago |
Megvii Engine Team
|
f6e66ec69c
|
fix(data): fix the sampling scale in Lighting
GitOrigin-RevId: d0a1ca50e2
|
4 years ago |
Megvii Engine Team
|
e18afa0b05
|
feat(mge/module): python wrapper for conv_transpose3d
GitOrigin-RevId: 61097b8713
|
4 years ago |
Megvii Engine Team
|
fdf7006ba7
|
fix(imperative): add __init__.py for tools
GitOrigin-RevId: 11ec38a4f1
|
4 years ago |
Megvii Engine Team
|
282dfc6232
|
refactor(imperative): alloc enum type class on heap
GitOrigin-RevId: d2b2acea22
|
4 years ago |
Megvii Engine Team
|
5c131d2a74
|
fix(device): fix get_device_count for xpu
GitOrigin-RevId: 6c77f507f3
|
4 years ago |
Megvii Engine Team
|
9eb4ba8836
|
perf(tensor): always copy to pinned memory before h2d
GitOrigin-RevId: f945f2435f
|
4 years ago |
Megvii Engine Team
|
4d9073fcba
|
test(dist): remove redundant assert for reduce_sum and gather
GitOrigin-RevId: bc7393bb63
|
4 years ago |
Megvii Engine Team
|
1a7112997c
|
feat(opr-mm): add backend argument for remote send/recv
GitOrigin-RevId: 841a0e45ab
|
4 years ago |
Megvii Engine Team
|
dee5a10af9
|
feat(distributed): auto detect device and backend when init group
GitOrigin-RevId: 90be2d5b4d
|
4 years ago |
Megvii Engine Team
|
c992afa960
|
test(mge/module): add module reset attribute test
GitOrigin-RevId: 6c9adc4a70
|
4 years ago |
Megvii Engine Team
|
93bfda5191
|
fix(mge/module): fix module attribute update mistake
GitOrigin-RevId: 842a8fcb6e
|
4 years ago |
Megvii Engine Team
|
33ab2454b9
|
fix(whl): support wheel version without sdk info,
which caused by python official PYPI do not support
LOCAL_VERSION; sync lite setup.py
GitOrigin-RevId: 7e35ceb5f1
|
4 years ago |
Megvii Engine Team
|
933188aa93
|
feat(functional/nn): support F.warp_perspective with `mat_idx`
GitOrigin-RevId: 66910c8bd8
|
4 years ago |
Megvii Engine Team
|
fe99cdc794
|
feat(interpreter): add dynamic sublinear
GitOrigin-RevId: 7de54fe743
|
4 years ago |
Megvii Engine Team
|
a81085221a
|
refactor(imperative): refactor enum param type caster
GitOrigin-RevId: 1aae07f143
|
4 years ago |
Megvii Engine Team
|
dcff115ef8
|
fix(distributed/launcher): fetch data early to avoid dead lock in launcher
GitOrigin-RevId: 9abcc956ef
|
4 years ago |
Megvii Engine Team
|
7e22e9f06e
|
feat(optimzer): add AdamW
GitOrigin-RevId: e608b5d5b9
|
4 years ago |