Megvii Engine Team
66f2dbd753
feat(mge/imperative): add module __repr__ for imperative
GitOrigin-RevId: ac13cc4659
4 years ago
Megvii Engine Team
4ab259f558
refactor(mge): add some deprecation and migration guides
GitOrigin-RevId: 4793adf02b
4 years ago
Megvii Engine Team
715009b564
refactor(mge/api): remove external, dropout fix
GitOrigin-RevId: 5e6ff1a372
4 years ago
Megvii Engine Team
a8c75ee530
refactor(mge/functional): make docs better
GitOrigin-RevId: acb287d48b
4 years ago
Megvii Engine Team
6c1dbd40d9
docs(mge): fix doctest
GitOrigin-RevId: 131fed8733
4 years ago
Megvii Engine Team
39d4328309
fix(mge/functional): functional api fixes
GitOrigin-RevId: fa206c4ff6
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
2beb65b19d
fix(mge/module): tensor shape will not work when constructing numpy array
GitOrigin-RevId: 5a0d705970
4 years ago
Megvii Engine Team
0067dcf068
test(mge/imperative): add module test to imperative
GitOrigin-RevId: e690bc42b0
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
6b380e8965
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
4 years ago