Megvii Engine Team
|
00ef677249
|
fix(mgb): remove internal for cambricon and atlas
GitOrigin-RevId: 861e349eb4
|
4 years ago |
Megvii Engine Team
|
134a102693
|
fix(mge/doc): fix typos, format
GitOrigin-RevId: 7d8037434e
|
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
|
f856b17085
|
fix(mge/functional): reshape bias to (1, out_features) in linear
GitOrigin-RevId: a15880b7fc
|
5 years ago |
Megvii Engine Team
|
08ac685e3a
|
feat(mge/functional): add logsumexp
GitOrigin-RevId: e7ef50e9ec
|
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
|
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
|
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 |
Megvii Engine Team
|
5eca4da323
|
feat(mge/functional): add softplus function
GitOrigin-RevId: 26c1ab7416
|
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
|
8659876794
|
docs(mge/functional): fix doctest
GitOrigin-RevId: cd23860ab8
|
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
|
2a1e0624ab
|
docs(mge/functional): fix doctest
GitOrigin-RevId: 45b6e1147e
|
5 years ago |
Megvii Engine Team
|
5d32db8a10
|
docs(mge/functional): fix docs error and refine description
GitOrigin-RevId: 62da6f0d60
|
5 years ago |
Megvii Engine Team
|
f71251fd0c
|
docs(mge/functional): refine the docstring of several apis
GitOrigin-RevId: 55f61be2a8
|
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
|
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
|
207527d108
|
fix(mge/clamp): fix `F.clamp`
GitOrigin-RevId: 1efac8add6
|
5 years ago |
Megvii Engine Team
|
e729006f8e
|
docs(mge/python): add doc for zero_grad
GitOrigin-RevId: bb5154c0e7
|
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
|
7ed98dd50a
|
feat(mge/functional): add conv_transpose2d support
GitOrigin-RevId: 9ad87a4ea9
|
5 years ago |
Megvii Engine Team
|
20332705d3
|
docs(mge/functional): refine tensor transpose doc
GitOrigin-RevId: 763d74d61e
|
5 years ago |
Megvii Engine Team
|
5178648422
|
fix(mge/doc): fix batch_norm2d doc
GitOrigin-RevId: 769ddd367b
|
5 years ago |
Megvii Engine Team
|
2c4d1afe14
|
test(mge): move `NUMBER` config to pytest.init
GitOrigin-RevId: 1d82209c40
|
5 years ago |
Megvii Engine Team
|
f33a92d635
|
docs(mge): fix api docs of loss and tranform
GitOrigin-RevId: 4d4d32b3c4
|
5 years ago |
Megvii Engine Team
|
e1b2d31d36
|
test(mge): skip some doctests having different results on gpu
GitOrigin-RevId: 66e5db5c89
|
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
|
8d2bbf7383
|
fix(mge/functional): support scalar inputs in elemwise functions
GitOrigin-RevId: 7bce561ee1
|
5 years ago |
Megvii Engine Team
|
d4f6a7bfed
|
fix(mge): fix tensor docstring
GitOrigin-RevId: 4dab166976
|
5 years ago |
Megvii Engine Team
|
e6d7c5e99a
|
docs(mge): fix docstring in loss and dataset
GitOrigin-RevId: 6b56673415
|
5 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
5 years ago |