Megvii Engine Team
|
0d8b913682
|
fix(mge/functional): reshape bias to (1, out_features) in linear
GitOrigin-RevId: a15880b7fc
|
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
|
5eca4da323
|
feat(mge/functional): add softplus function
GitOrigin-RevId: 26c1ab7416
|
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
|
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
|
7ed98dd50a
|
feat(mge/functional): add conv_transpose2d support
GitOrigin-RevId: 9ad87a4ea9
|
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
|
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
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
5 years ago |