Megvii Engine Team
4e95c13617
feat(sgd): sgd supports nesterov momentum
GitOrigin-RevId: 13eda179da
3 years ago
Megvii Engine Team
9451a96147
test(mge/optimizer): update optimizer test to make sure grad not change
GitOrigin-RevId: e207672116
4 years ago
Megvii Engine Team
7e22e9f06e
feat(optimzer): add AdamW
GitOrigin-RevId: e608b5d5b9
4 years ago
Megvii Engine Team
31e4bf2c34
fix(mge/optimizer): fix optimizer's state_dict bug
GitOrigin-RevId: 67fb112fb8
4 years ago
Megvii Engine Team
30d6b4f65d
fix(mge): fix scalar parameter change to 1-dim parameter after backward and optimize
GitOrigin-RevId: 1794369a71
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
c294b9d18b
refactor(mge/tensor): remove old implementation
remove core.tensor, raw_tensor,TensorWrapper
avoid create tensor with zero-stride numpy ndarray
GitOrigin-RevId: 4fe5c4c5ba
4 years ago
Megvii Engine Team
638ab52fdc
feat(mge/imperative): simulates scalar
GitOrigin-RevId: e81630e256
4 years ago
Megvii Engine Team
cc85047bf0
fix(mge/trace): fix sublinear in trace
GitOrigin-RevId: 356dcd9523
4 years ago
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
087ceb52ef
feat(mge/imperative): add more optimizer trace tests
GitOrigin-RevId: 4127de1d22
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