Megvii Engine Team
|
f386381080
|
fix(imperative): fix inplace operation of optim
GitOrigin-RevId: 2aaa71eb66
|
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 |
Megvii Engine Team
|
261a5bce23
|
feat(imperative/amp): add dimshuffle in set_format for nhwc
GitOrigin-RevId: 5ced9e1a31
|
3 years ago |
Megvii Engine Team
|
cd26376549
|
style(imperative/amp): reformat code
GitOrigin-RevId: 6e5a6e1eaf
|
3 years ago |
Megvii Engine Team
|
fc633ce4ff
|
fix(imperative/amp): fix custom grad in Subgraph
GitOrigin-RevId: 1c728d6ab9
|
3 years ago |
Megvii Engine Team
|
fc0f454685
|
fix(dnn/check_non_finite): adjust some details of CheckNonFinite
GitOrigin-RevId: 52ddd805b4
|
3 years ago |
Megvii Engine Team
|
e393d1cf65
|
feat(mge/amp): add convert_format module for NHWC training
GitOrigin-RevId: 1b41e1042c
|
3 years ago |
Megvii Engine Team
|
533fb5bf49
|
feat(imperative): support formatted tensor and add special op rules
GitOrigin-RevId: 77ff909f23
|
3 years ago |
Megvii Engine Team
|
c2e9860feb
|
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
|
3 years ago |
Megvii Engine Team
|
3c3fc6f33c
|
refactor(imperative): move python code of elemwise/reduce/conv2d/bn to c++
GitOrigin-RevId: 01b5324392
|
3 years ago |
Megvii Engine Team
|
11d75fecb5
|
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
|
3 years ago |
Megvii Engine Team
|
5cef74a77e
|
feat(mge/amp): add GradScaler support
GitOrigin-RevId: 0ab4910360
|
4 years ago |
Megvii Engine Team
|
1bf18252c4
|
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
|
4 years ago |