Megvii Engine Team
10e942d952
refactor(mge): polish api
- refactor(mge): add support for optimizer.step().clear_grad() idiom
- refactor(mge): rename some methods of GradManager
- refactor(mge): remove tensor_nn and TensorDict
- refactor(mge): remove Buffer
- refactor(mge): remove requires_grad flag
- refactor(mge): add a default grad=None attribute to Tensor
- refactor(mge): deprecation for 1.0
GitOrigin-RevId: 3b723d9387
4 years ago
Megvii Engine Team
66b6daf777
test(mge/optimizer): fix test for new optimizer api
GitOrigin-RevId: 482ee62652
4 years ago
Megvii Engine Team
e283663a02
fix(mge/imperative): update tests to new optimizer api
GitOrigin-RevId: 3d06e3db3c
4 years ago
Megvii Engine Team
fc3eca845b
feat(mge/imperative/jit): add dump input shape and xornet example
GitOrigin-RevId: 5e2acd4052
4 years ago
Megvii Engine Team
087ceb52ef
feat(mge/imperative): add more optimizer trace tests
GitOrigin-RevId: 4127de1d22
4 years ago
Megvii Engine Team
0e82b959a1
feat(mge/imperative): add sublinear options
GitOrigin-RevId: f0e917f716
4 years ago
Megvii Engine Team
afddefb677
feat(mge/imperative): add more trace test
GitOrigin-RevId: b02e420a8a
4 years ago
Megvii Engine Team
e1fba6ece7
test(mge/distributed): add get_device_count_by_fork to fix distributed test skip
GitOrigin-RevId: 9ffd8a6149
4 years ago
Megvii Engine Team
6e882c1a86
feat(whl/imperative): compat for build python whl imperative and legacy runtime
GitOrigin-RevId: 7f6629ae1f
4 years ago
Megvii Engine Team
40d18c8920
fix(mge/imperative): fix tests when shape is tensor
GitOrigin-RevId: fd0095c1ec
4 years ago
Megvii Engine Team
227ec04bd3
fix(mge/test): fix pytest failed issue
GitOrigin-RevId: 96df470406
4 years ago
Megvii Engine Team
6b380e8965
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
4 years ago