Megvii Engine Team
|
d9a9d9d49e
|
fix(imperative/fastrun): set workspace limit for imperative rt
GitOrigin-RevId: 474dc691a3
|
3 years ago |
Megvii Engine Team
|
ac86d64474
|
fix(mge/module): fix prelu error when use_symbolic_shape is true
GitOrigin-RevId: 25b9c4d41d
|
3 years ago |
Megvii Engine Team
|
4f6c5d8f01
|
feat(mge/dump): enable jit.dump to dump with testcase
GitOrigin-RevId: 5dce356452
|
3 years ago |
Megvii Engine Team
|
182ca25d45
|
feat(mge/imperative): add coalesce_all_free_memory mge api
GitOrigin-RevId: ea05b1e9a1
|
3 years ago |
Megvii Engine Team
|
9d05738d54
|
Merge pull request #225 from chenjiahui0131:pixel_shuffle_module
GitOrigin-RevId: d94533655a
|
3 years ago |
Megvii Engine Team
|
936bb23738
|
fix(mge/tensor): fix is/is not SyntaxWarning occurs in python3.8
GitOrigin-RevId: 12b66da367
|
3 years ago |
Megvii Engine Team
|
8d507cc39d
|
docs(mge/module): add note about definition of momentum in BatchNorm
GitOrigin-RevId: 30a9aec20f
|
3 years ago |
Megvii Engine Team
|
676b205bd8
|
style(imperative): add bypass of clang-format for cg options
GitOrigin-RevId: 824d610631
|
3 years ago |
Megvii Engine Team
|
2c5a007a92
|
fix(mge/optimizer): allow lr to be 0
GitOrigin-RevId: dabd1fcc33
|
3 years ago |
chenjiahui
|
c8a9a3fd0a
|
feat(imperative): add PixelShuffle to module.__init__.py
|
3 years ago |
Megvii Engine Team
|
fe15239ac0
|
fix(imperative): fix error message for tensors with intbx data type
GitOrigin-RevId: cbb42f8127
|
3 years ago |
Megvii Engine Team
|
feb813bcc0
|
chore(release): bump version
GitOrigin-RevId: 1da5b93cae
|
3 years ago |
Megvii Engine Team
|
29f9935d92
|
fix(imperative/python): add layer_norm doc and rm useless param
GitOrigin-RevId: 1b15db621e
|
3 years ago |
Megvii Engine Team
|
300955140e
|
fix(traced_module/doc): fix warning for traced_module docstring
GitOrigin-RevId: 5d2d047d2f
|
3 years ago |
Megvii Engine Team
|
f7d05db7d6
|
fix(mge/module): fix named_tensors
GitOrigin-RevId: bb5aa1f41d
|
3 years ago |
Megvii Engine Team
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
3 years ago |
Megvii Engine Team
|
bfb30dcb81
|
chore(format): fix compile bugs after code format
GitOrigin-RevId: 11a4b06f6f
|
3 years ago |
Megvii Engine Team
|
2108696588
|
feat(mge/imperative): add local response normalization
GitOrigin-RevId: 939a4d26dd
|
3 years ago |
Megvii Engine Team
|
ca4c93dee7
|
fix(mge): fix layer norm amp bug
GitOrigin-RevId: dba691fcbf
|
3 years ago |
Megvii Engine Team
|
c674bf0e6e
|
feat(debug): add gdb-scripts
GitOrigin-RevId: 43f9674f50
|
3 years ago |
Megvii Engine Team
|
177dec94c5
|
feat(mgb/opr): add bgr2gray mode for cvtcolor opr
GitOrigin-RevId: d50415b236
|
3 years ago |
Megvii Engine Team
|
000517c641
|
fix(grad): stop using exception in grad_override
GitOrigin-RevId: 00ae38d48b
|
3 years ago |
Megvii Engine Team
|
d02b0a08f7
|
feat(lite): fix typo
GitOrigin-RevId: 8c46aa3a30
|
3 years ago |
Megvii Engine Team
|
cebda6ff8d
|
feat(mge/imperative): add ctc loss
GitOrigin-RevId: e29854a98e
|
3 years ago |
Megvii Engine Team
|
f5cb21ed3a
|
fix(mgb/opr): add non finite check
GitOrigin-RevId: a9fcd0a350
|
3 years ago |
Megvii Engine Team
|
b6142bee9a
|
feat(imperative): support tensor with uint16 date type
GitOrigin-RevId: 57ba0633c7
|
3 years ago |
Megvii Engine Team
|
4e95c13617
|
feat(sgd): sgd supports nesterov momentum
GitOrigin-RevId: 13eda179da
|
3 years ago |
Megvii Engine Team
|
dc28a43409
|
fix(mgb/bn): fix empty tensor input problem and other minor problems
GitOrigin-RevId: 1708b76cb8
|
3 years ago |
Megvii Engine Team
|
270f1aa28f
|
feat(mgb/serialization): add Accessor for OprLoader to fix BN output compatibility
GitOrigin-RevId: 3b95da02c8
|
3 years ago |
Megvii Engine Team
|
c0ccd0ea7e
|
feat(mge/bn): add NHWC support for bn
GitOrigin-RevId: 0a5bb6f72d
|
3 years ago |
chenjiahui
|
6b863cc53a
|
feat(imperative): add pixel_shuffle opr
Closes #211
GitOrigin-RevId: 1846714791
|
3 years ago |
Megvii Engine Team
|
b982be567a
|
feat(mge/imperative): add permutation support for the tensor
GitOrigin-RevId: 7ed0447bfe
|
3 years ago |
Megvii Engine Team
|
3977b7aa0b
|
feat(mgb/shuffle): add shuffle opr
GitOrigin-RevId: 80490a6f84
|
3 years ago |
Megvii Engine Team
|
884a07ffd2
|
fix(test/random): set a random seed for random unit test
GitOrigin-RevId: ad4b01eac7
|
3 years ago |
Megvii Engine Team
|
3f0bb47a43
|
ci(imperative): add test requires
GitOrigin-RevId: d7a945caf3
|
3 years ago |
Megvii Engine Team
|
11a383bc79
|
feat(mgb): add tensorboard tool c++ layer interface and python parsing
GitOrigin-RevId: ae98bc887f
|
3 years ago |
Megvii Engine Team
|
8084e4e2ee
|
feat(mgb): add tensorboard tool python layer interface
GitOrigin-RevId: 065bc4d153
|
3 years ago |
Megvii Engine Team
|
97b1b7774d
|
feat(mgb): add megbrain layer norm opr with subgraph
GitOrigin-RevId: 9b7fa821f8
|
3 years ago |
Megvii Engine Team
|
65d554ed8e
|
refactor(dtr): improve dtr in computing graph
GitOrigin-RevId: d599a08e52
|
3 years ago |
Megvii Engine Team
|
f642b05e17
|
test(mge): update traced_module unit test
GitOrigin-RevId: 3948d50d79
|
3 years ago |
Megvii Engine Team
|
fb20cb36a0
|
docs(mge/traced_module): update traced_module api doc
GitOrigin-RevId: 19a95d26c7
|
3 years ago |
Megvii Engine Team
|
c7a8d945c7
|
fix(mge/traced_module): let graph record total_id
GitOrigin-RevId: f99178f3ac
|
3 years ago |
Megvii Engine Team
|
4de62ad6cd
|
docs(mge/device): fix docstring of get_cuda_compute_capability
GitOrigin-RevId: 4ec04e8b9b
|
3 years ago |
Megvii Engine Team
|
fb49a2834f
|
refactor(mgb/dnn): refactor enum used in serializing
GitOrigin-RevId: e57af4a59c
|
3 years ago |
Megvii Engine Team
|
6082c353e7
|
feat(dnn/rocm): support bool in type_cvt and elemwise
GitOrigin-RevId: ad5ec7bc1c
|
4 years ago |
Megvii Engine Team
|
93f70a9539
|
feat(src): add MGB_CUSTOM_OP macro
GitOrigin-RevId: beddcf9745
|
3 years ago |
Megvii Engine Team
|
b3e958d0bc
|
fix(src): fix the warnings and copy.bara.sky in custom op
GitOrigin-RevId: 4ade455897
|
3 years ago |
Megvii Engine Team
|
cdb692d2fa
|
refactor(imperative): add TODO tag for some functions
GitOrigin-RevId: e295a1fa55
|
3 years ago |
Megvii Engine Team
|
90dd07161c
|
refactor(imperative): modify the python interface of custom op
GitOrigin-RevId: e82e5de480
|
3 years ago |
Megvii Engine Team
|
cbf024bf20
|
feat(imperative): add adaptor between custom op and imperative runtime
GitOrigin-RevId: d7877f2e32
|
3 years ago |