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
44d0b5daf5
feat(imperative): enable to() to copy to device
GitOrigin-RevId: f9caf17d24
4 years ago
Megvii Engine Team
939bf3b28d
fix(mge/imperative): impl hashable for nms and cond_take
GitOrigin-RevId: 56918db014
4 years ago
Megvii Engine Team
8dc23e0fdf
fix(mge/functional): fix indexing_one_hot and remote_recv
GitOrigin-RevId: 00bdfb502b
4 years ago
Megvii Engine Team
03320a0548
fix(functional): change to user-friendly nms api and fix concat
GitOrigin-RevId: 76b4190270
4 years ago
Megvii Engine Team
7fadc16d3c
refactor(mge/functional): support tensor shape in interpolate and split
GitOrigin-RevId: 6430b64f01
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
60076f4715
test(mge/parampack): add parampack related opr test
GitOrigin-RevId: 550c6b6443
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
f4927db2fb
feat(mge/functional): support where func
GitOrigin-RevId: 9df6421ebe
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