Megvii Engine Team
|
a8b146d74b
|
Merge pull request #397 from kagome112:master
GitOrigin-RevId: e40b12f76e
|
2 years ago |
Megvii Engine Team
|
5655636c1c
|
chore(mge/metric): funtional.topk_acc (add nn) use functional.metric
GitOrigin-RevId: 77fd432cb2
|
2 years ago |
Megvii Engine Team
|
d5688c3d38
|
chore(mge/functional): deprecate debug_param and prefer config
GitOrigin-RevId: 0d7126641b
|
2 years ago |
Megvii Engine Team
|
9bb2e395ce
|
chore(mge/tensorcache): rename the module making it not public
GitOrigin-RevId: 559d205a8c
|
2 years ago |
Megvii Engine Team
|
6e56d1bd57
|
chore(mge/deprecated): add __deprecated__ attr for condition case
GitOrigin-RevId: 6d8e7398e4
|
2 years ago |
Megvii Engine Team
|
98460f583f
|
chore(mge/misc): rename APIs that should not be public
GitOrigin-RevId: b72517bfe9
|
2 years ago |
Megvii Engine Team
|
b2959589fc
|
docs(mge): update AvgPool2d example
GitOrigin-RevId: 2d5e515055
|
2 years ago |
Megvii Engine Team
|
ada54ba5a0
|
docs(mge): update AvgPool2d & functional.nn.avg_pool2d example
GitOrigin-RevId: e2859c7f99
|
2 years ago |
Megvii Engine Team
|
d8ea01680c
|
docs(mge): update AvgPool2d & functional.nn.max_pool2d docstring
GitOrigin-RevId: d69c5b024f
|
2 years ago |
Megvii Engine Team
|
01353b2d90
|
Merge pull request #442 from Seeker98:avgpool_doc
GitOrigin-RevId: d048955398
|
2 years ago |
Megvii Engine Team
|
25f97b76ea
|
feat(imperative): load_nerwork_and_run enable weight preprocess
GitOrigin-RevId: 0642b237a7
|
2 years ago |
Megvii Engine Team
|
99309fa3d6
|
feat(mge/functional): add param output_padding for deconv ops
GitOrigin-RevId: 8a69608953
|
2 years ago |
Megvii Engine Team
|
b8c7557b86
|
fix(mm): fix mm error when use sync
GitOrigin-RevId: 63387bda04
|
2 years ago |
Megvii Engine Team
|
2797fcfad0
|
fix(mge/device): add missed API to __all__ scope
GitOrigin-RevId: c3ce90990f
|
2 years ago |
Megvii Engine Team
|
fac67e7c2b
|
feat(gopt): support nchw44 global pooling with fuse_grain
GitOrigin-RevId: 4c43a149f8
|
2 years ago |
Megvii Engine Team
|
32b31fd578
|
fix(mgb): change the check method of cuda sm code
GitOrigin-RevId: 23dbc9b574
|
2 years ago |
Megvii Engine Team
|
5f86368219
|
Revert "feat(dnn): add elemwise modes"
This reverts commit cb713ddb24 .
GitOrigin-RevId: 02adf025e6
|
2 years ago |
Megvii Engine Team
|
d2a1905ad5
|
Revert "feat(mgb): add cumprod opr"
This reverts commit 3436c3bdaa .
GitOrigin-RevId: 95ab3d1aa7
|
2 years ago |
Megvii Engine Team
|
49e14f87b5
|
feat(mgb): add cumprod opr
GitOrigin-RevId: 3436c3bdaa
|
3 years ago |
Megvii Engine Team
|
87aedc2991
|
feat(dnn): add elemwise modes
GitOrigin-RevId: cb713ddb24
|
3 years ago |
Megvii Engine Team
|
cc31b9db34
|
docs(mge/functional): fix vision docs
GitOrigin-RevId: f03567d8ae
|
2 years ago |
Megvii Engine Team
|
6f7649e935
|
docs(docstring): fix pad docstring
GitOrigin-RevId: 698d2f1c0d
|
2 years ago |
Megvii Engine Team
|
bf32d1e04f
|
docs(dataloader): update dataloader docstring
GitOrigin-RevId: 3e94a4bdf4
|
2 years ago |
Megvii Engine Team
|
4a32cc493a
|
docs(mge/data): update Dataset class docstring
GitOrigin-RevId: f08d818cf3
|
2 years ago |
Megvii Engine Team
|
70fc568224
|
docs(mge/data): update MNIST dataset docstring
GitOrigin-RevId: 536a46325f
|
2 years ago |
Megvii Engine Team
|
d60d028a40
|
feat(mge/device): enable to get cuda/cudnn/tensorrt version
GitOrigin-RevId: 5864c61d10
|
3 years ago |
Megvii Engine Team
|
19af26887e
|
fix(mge/tools): fix module_stats for duplicated module
GitOrigin-RevId: a15f17d616
|
2 years ago |
Megvii Engine Team
|
4cd4a38a7d
|
fix(mge/tools): fix network_visualize for op without out shapes
GitOrigin-RevId: fdde52c214
|
2 years ago |
Megvii Engine Team
|
51397bfc08
|
feat(mgb): supports value infer and empty input tensor in ElemwiseMultiType
GitOrigin-RevId: 05577a8bc8
|
3 years ago |
Megvii Engine Team
|
247e2f59a4
|
feat(mgb/dnn): add modes that the output type is bool in elemwise
GitOrigin-RevId: fd0134fca2
|
3 years ago |
Megvii Engine Team
|
f386381080
|
fix(imperative): fix inplace operation of optim
GitOrigin-RevId: 2aaa71eb66
|
3 years ago |
Megvii Engine Team
|
6c4c4ca6e6
|
Merge pull request #459 from Qsingle:fix_overflow_of_flops_calculate
GitOrigin-RevId: c1333e2089
|
3 years ago |
Megvii Engine Team
|
ac51f7807d
|
feat(mge/distributed): add support for batch send recv op
GitOrigin-RevId: eb3d712704
|
3 years ago |
Megvii Engine Team
|
5ba7663732
|
fix(traced_module): fix trace_module function may raise error in finally scope
GitOrigin-RevId: 2689b1f6e6
|
3 years ago |
Megvii Engine Team
|
e6dcfbe806
|
fix(traced_module): fix traced module compatible issues
GitOrigin-RevId: 67e68ef5ea
|
3 years ago |
Megvii Engine Team
|
18f83a258e
|
docs(mge/functional): fix F.svd docstring
GitOrigin-RevId: b84e5fdc46
|
3 years ago |
Megvii Engine Team
|
a49d5bf973
|
fix(autodiff): fix inplace operation on autodiff.Function
GitOrigin-RevId: a658680f35
|
3 years ago |
Megvii Engine Team
|
2484cd2741
|
fix(tensor): check args when construct tensor with existing tensor
GitOrigin-RevId: 0345454070
|
3 years ago |
Megvii Engine Team
|
657db8dce2
|
chore(tools): remove dump_with_testcase_mge, user should use jit.dump instead
GitOrigin-RevId: a88db6e475
|
3 years ago |
Megvii Engine Team
|
24c5c19bf0
|
fix(imperative): make functional ops support negative axis
GitOrigin-RevId: f61e01270b
|
3 years ago |
Megvii Engine Team
|
f96429c031
|
feat(imperative): support empty tensor in roi_align
GitOrigin-RevId: aeb2770401
|
3 years ago |
Megvii Engine Team
|
07bdb3bf1e
|
feat(imperative): add swapaxes
GitOrigin-RevId: e84014a011
|
3 years ago |
Megvii Engine Team
|
df5ebd3da7
|
fix(imperative/ops): fix the vmemory problem in 1.9
GitOrigin-RevId: 728e983701
|
3 years ago |
Megvii Engine Team
|
b36b5bd8cb
|
refactor(mgb): check input when profiling
GitOrigin-RevId: 1d722dd741
|
3 years ago |
Megvii Engine Team
|
eef0308b29
|
feat(serialization): add no_change_graph and version param whem dump model
GitOrigin-RevId: 65064452c9
|
3 years ago |
Megvii Engine Team
|
4ab5f970e9
|
fix(build): fix ci error
GitOrigin-RevId: 9cbf64dda2
|
3 years ago |
Megvii Engine Team
|
b9a6932341
|
feat(imperative): channel default model format to fbs v2
GitOrigin-RevId: 6066516c31
|
3 years ago |
Megvii Engine Team
|
da91e650a5
|
refactor(ops/layer_norm): speed up the host speed of layer_norm
GitOrigin-RevId: 6f359b5b29
|
3 years ago |
Megvii Engine Team
|
67cfce9f2f
|
fix(imperative/amp): add is_scalar check in elemwise and concat
GitOrigin-RevId: 61a612e92a
|
3 years ago |
Megvii Engine Team
|
d313f92610
|
fix(imperative/amp): fix format transformation for symbol trans
GitOrigin-RevId: 96cc237c67
|
3 years ago |