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
a6f456415f
refactor(mge/dtype): modify some interface name and enrich comments
GitOrigin-RevId: f9217f6d27
5 years ago
Megvii Engine Team
d3730036ea
test(mge/dist): add unit tests for megengine.distributed.utils
GitOrigin-RevId: 32336c66f1
5 years ago
Megvii Engine Team
ffe0409398
docs(mge/optimizer): refine the docstring of several apis
GitOrigin-RevId: a97fe5b68a
5 years ago
Megvii Engine Team
0c54f2dc13
docs(mge/core): refine the docstring of several apis
GitOrigin-RevId: c03fecfa6a
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
8cf7150df0
feat(dnn/mge): add compnode multithread in python
GitOrigin-RevId: 47373d291d
5 years ago
Megvii Engine Team
4997270156
docs(mge/module): refine the docstring of several apis
GitOrigin-RevId: ea04e05be4
5 years ago
Megvii Engine Team
4c5d01fa34
refactor(mge/examples): add example to use modified sublinear API
GitOrigin-RevId: c6f6c65d0b
5 years ago
Megvii Engine Team
396653df5f
feat(mge/api): expose sublinear related parameters at mge api level
GitOrigin-RevId: 7a47f5d0d5
5 years ago
Megvii Engine Team
d8d3f405bb
fix(mge/data): fix an nan bug in coco dataset
GitOrigin-RevId: 689fbaafa5
5 years ago
Megvii Engine Team
41464e1818
feat(mge/data): voc dataset supports detection
GitOrigin-RevId: f78bef3cd2
5 years ago
Megvii Engine Team
26738d99eb
feat(mge/data): add objects365 dataset
GitOrigin-RevId: 8eccebfc1d
5 years ago
Megvii Engine Team
aa2bfd2d85
fix(mge/data): create dir automatically when download=True
GitOrigin-RevId: f3cb1b7d50
5 years ago
Megvii Engine Team
b8cbd4510b
fix(mgb/mm_server): return -1 when create_mm_server failed
GitOrigin-RevId: 117bb80c06
5 years ago
Megvii Engine Team
abf82cfb50
feat(mge/dtype): export `_metadata_dict` for consistent dtype property
GitOrigin-RevId: 6840c0b6b4
5 years ago
Megvii Engine Team
dd962a4700
fix(mge/module): add negative slope attribute for LeakyReLU module
GitOrigin-RevId: 67e1967b56
5 years ago
Megvii Engine Team
554ce352c8
feat(mgb/gopt): add nchw44 optpass
GitOrigin-RevId: dc38724558
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
6ffcfb4ce2
refactor(mge/dataloader): make plasma_store starting more robust
GitOrigin-RevId: e1f78db99a
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
99fae95e02
feat(mge/parampack): add user-defined key to pack params
GitOrigin-RevId: 7d51dcae23
5 years ago
Megvii Engine Team
9a42c63641
fix(mge/module): add the more warnings in load_state_dict
In the before, the information of miss matching and unused operators won't be printed in the non-trict mode, this commit add the information.
GitOrigin-RevId: b2543eb832
5 years ago
Megvii Engine Team
9b74e6fc88
fix(mge/random): add one assert in distribution.py
low must be less than high in uniform sampling
GitOrigin-RevId: 34c33a1ac9
5 years ago
Megvii Engine Team
fc4dedc0bc
fix(mge/random): add lower bound and higher bound for uniform sampling
GitOrigin-RevId: 2a2c56fd17
5 years ago
Megvii Engine Team
779ed45d74
docs(mge/parampack): refine parampack doc
GitOrigin-RevId: 1cff83c308
5 years ago
Megvii Engine Team
8e8794040d
chore(release): bump version.
GitOrigin-RevId: 979d8c247b
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
b588d93eb0
fix(mge): fix fast-run cache mismatch
GitOrigin-RevId: a41150f71a
5 years ago
Megvii Engine Team
da52256891
fix(mge/data): process mnist without generate new files
GitOrigin-RevId: 44a697c3fe
5 years ago
Megvii Engine Team
5178648422
fix(mge/doc): fix batch_norm2d doc
GitOrigin-RevId: 769ddd367b
5 years ago
Megvii Engine Team
f077a5292c
fix(mge/param_pack): release old parameters
GitOrigin-RevId: 40a1f044e9
5 years ago
Megvii Engine Team
48822557b8
docs(mge/bn): fix docs and tests of batchnorm
GitOrigin-RevId: 8a96aa5fc2
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
fc6aa12e2f
feat(mge):add qint4/quint4 to python
GitOrigin-RevId: f94609db00
5 years ago
Megvii Engine Team
2c2caf3331
refactor(mge/data/dataloader): refactor the implementation of parallel dataloader
GitOrigin-RevId: 0554ee8427
5 years ago
Megvii Engine Team
364dafccf6
fix typo
GitOrigin-RevId: e1c3287467
5 years ago
Megvii Engine Team
94a7696924
refactor ImageNet
GitOrigin-RevId: f7774e0ffc
5 years ago
Megvii Engine Team
e677ffc85d
fix(mge/dataset) fix imagenet dir error
GitOrigin-RevId: e0a59464bb
5 years ago
Megvii Engine Team
8d2bbf7383
fix(mge/functional): support scalar inputs in elemwise functions
GitOrigin-RevId: 7bce561ee1
5 years ago