Megvii Engine Team
|
cfbd4f9bf1
|
fix(device): fix get_device_count for xpu
GitOrigin-RevId: 6c77f507f3
|
4 years ago |
Megvii Engine Team
|
04b96602ae
|
feat(opr-mm): add backend argument for remote send/recv
GitOrigin-RevId: 841a0e45ab
|
4 years ago |
Megvii Engine Team
|
d4eb389042
|
feat(distributed): auto detect device and backend when init group
GitOrigin-RevId: 90be2d5b4d
|
4 years ago |
Megvii Engine Team
|
7a9d1f57cd
|
fix(mge/module): fix module attribute update mistake
GitOrigin-RevId: 842a8fcb6e
|
4 years ago |
Megvii Engine Team
|
976b351ada
|
feat(functional/nn): support F.warp_perspective with `mat_idx`
GitOrigin-RevId: 66910c8bd8
|
4 years ago |
Megvii Engine Team
|
899d70a8fa
|
fix(imperative): add __init__.py for tools
GitOrigin-RevId: c260c572f2
|
4 years ago |
Megvii Engine Team
|
fe99cdc794
|
feat(interpreter): add dynamic sublinear
GitOrigin-RevId: 7de54fe743
|
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 |
Megvii Engine Team
|
2d18074a70
|
fix(mgb): fix spell error
GitOrigin-RevId: acae00e0a5
|
4 years ago |
Megvii Engine Team
|
65af9cfcdc
|
refactor(mge): use lower case for default string parameters in functional and module
GitOrigin-RevId: dbc1f27ff7
|
4 years ago |
Megvii Engine Team
|
d0aa9b41ee
|
refactor(mge/functional): move nvof to vision, compatible with old usage
GitOrigin-RevId: cd9d9b4f5a
|
4 years ago |
Megvii Engine Team
|
b1baee60f5
|
feat(imperative/utils): add optimize-for-inference interface for opgraph
GitOrigin-RevId: 9f93f82190
|
4 years ago |
Megvii Engine Team
|
86598c8227
|
fix(mge/utils): filter out parameter "arg_names" and "output_name" in network.dump
GitOrigin-RevId: 408f52ad2d
|
4 years ago |
Megvii Engine Team
|
9dbe71dd1a
|
feat(mge/utils): add array method for varnode
GitOrigin-RevId: 6e4d05b475
|
4 years ago |
Megvii Engine Team
|
9b0bd69521
|
fix(imperative/utils): fix name filter of network varnode
GitOrigin-RevId: 20ead35335
|
4 years ago |
Megvii Engine Team
|
af62cf6f05
|
fix(mge/utils): fix auto naming bug when expanding structure
GitOrigin-RevId: 5735bf5151
|
4 years ago |
Megvii Engine Team
|
4f0e6eae35
|
fix(mge/trace): re-open gopt level in trace
GitOrigin-RevId: 5ebc712690
|
4 years ago |
Megvii Engine Team
|
da167cbc05
|
fix(mge/utils): fix module stats calculate flops bug for group conv and remove model status change
GitOrigin-RevId: 647dc6eb66
|
4 years ago |
Megvii Engine Team
|
6bb9a2556f
|
fix(mge/tools): fix module stats' receptive field bug for Module
GitOrigin-RevId: b471363830
|
4 years ago |
Megvii Engine Team
|
44bafd3f58
|
fix(imperative/quantization): fix zero scale bug of easy quant
GitOrigin-RevId: f45e19b3e4
|
4 years ago |
Megvii Engine Team
|
e474994f46
|
feat(imperative/jit): catch input tensors name when tracing
GitOrigin-RevId: 9c69254866
|
4 years ago |
Megvii Engine Team
|
aed681d325
|
feat(imperative/utils): optimize the naming rules
GitOrigin-RevId: 329bac640a
|
4 years ago |
Megvii Engine Team
|
c6bbc478c7
|
feat(imperative/utils): add __repr__ for network node
GitOrigin-RevId: fe5b59749f
|
4 years ago |
Megvii Engine Team
|
2df847544d
|
fix(mge/tools): fix node display bug in tensorboard
GitOrigin-RevId: c997d6cccb
|
4 years ago |
Megvii Engine Team
|
13481fd2ca
|
feat(mge/tools): add support of receptive_field stats for NetworkNode
GitOrigin-RevId: 11ef335468
|
4 years ago |
Megvii Engine Team
|
5a7c30e0f7
|
feat(mge/tools): add summary print for module_stats and network_visualize
GitOrigin-RevId: 7d85aa0ea2
|
4 years ago |
Megvii Engine Team
|
3103539175
|
feat(mge/tools): set network_visualize's log_path as optional flag
GitOrigin-RevId: a74bdc08ba
|
4 years ago |
Megvii Engine Team
|
0ad85a41ac
|
test(mge): do not ignore data test
GitOrigin-RevId: 937d24df5e
|
4 years ago |
Megvii Engine Team
|
31e4bf2c34
|
fix(mge/optimizer): fix optimizer's state_dict bug
GitOrigin-RevId: 67fb112fb8
|
4 years ago |
Megvii Engine Team
|
dc96f6aa49
|
fix(mge/quantization): fix quantized concat forward problem
GitOrigin-RevId: dc21b340d1
|
4 years ago |
Megvii Engine Team
|
48a7d333c3
|
fix(mge/module): fix auto-naming error when there are containers in the module
GitOrigin-RevId: 6b8f043321
|
4 years ago |
Megvii Engine Team
|
702ed9eee6
|
feat(mge): enable cg dump for debugging
GitOrigin-RevId: 4292364d0d
|
4 years ago |
Megvii Engine Team
|
37c3e0a953
|
chore(mge): remove old version.py and use version_template
|
4 years ago |
Megvii Engine Team
|
e9982d613f
|
chore(mge/functional): add compatible code for functional api
GitOrigin-RevId: 3b2f829cc5
|
4 years ago |
Megvii Engine Team
|
4917534b65
|
feat(mge/optimizer): save state's numpy value by default in `state_dict`
GitOrigin-RevId: ec7e4d56f5
|
4 years ago |
Megvii Engine Team
|
84f990a051
|
fix(mge/tensor): fix `__repr__` for Tensor's subclass
GitOrigin-RevId: 91b58e6dca
|
4 years ago |
Megvii Engine Team
|
cdbb4a20ff
|
fix(mge/tensor): fix tensor's serialization behavior
GitOrigin-RevId: 4d74a4b46e
|
4 years ago |
Megvii Engine Team
|
a3ea1f153c
|
feat(mgb/opr): add fast profile and combined Execution strategy
GitOrigin-RevId: 843dc3a790
|
4 years ago |
Megvii Engine Team
|
992a90bbad
|
docs(mge/quantization): add docstring for Observer
GitOrigin-RevId: 043be3886d
|
4 years ago |
Megvii Engine Team
|
c45f1eb298
|
fix(mge/tools): improve `module_visualize` result's robustness and beauty
GitOrigin-RevId: ef7b573776
|
4 years ago |
Megvii Engine Team
|
786c36ff42
|
fix(mge/tools): rename `net_stats` in function and examples to match file name
GitOrigin-RevId: 82a1377d66
|
4 years ago |
Megvii Engine Team
|
9c90ce8c33
|
fix(mge/quantization): fix `quantize` and `quantize_qat`'s `set_expand_structure` arguments
GitOrigin-RevId: c61633095d
|
4 years ago |
Megvii Engine Team
|
c8697a7005
|
feat(imperative/src): python wrapper for cambricon and atlas runtime opr
GitOrigin-RevId: bd969d1339
|
4 years ago |
Megvii Engine Team
|
8cfed4a1cb
|
refactor(functional/nn): remove nn.svd
GitOrigin-RevId: 4f58e10a1f
|
4 years ago |
Megvii Engine Team
|
e6b06914f7
|
feat(mge/tensor): support non-Tensor value in `_reset` and remove depreciated tests
GitOrigin-RevId: faf6c78aa8
|
4 years ago |
Megvii Engine Team
|
5a38ad3920
|
feat(mge/utils): add get/set_expand_structure to deal with complex key
GitOrigin-RevId: 4d1b952068
|
4 years ago |
Megvii Engine Team
|
fad5bc74c3
|
feat(mge/tools): add load_network_and_run for megengine
GitOrigin-RevId: 7d8ed5af79
|
4 years ago |
Megvii Engine Team
|
3f2e15f534
|
fix(mge/logger): fix set_mgb_log_level
GitOrigin-RevId: 6c94c51483
|
4 years ago |
Megvii Engine Team
|
594fa722bd
|
feat(mge/graph): add `modify_opr_algo_strategy_inplace` for fast-run
GitOrigin-RevId: 034cf58b2a
|
4 years ago |