Megvii Engine Team
84d99d1cc4
fix(traced_module): fix Module compatible issue and traced module getattr check
GitOrigin-RevId: 62eb3bfb10
3 years ago
Megvii Engine Team
f7d05db7d6
fix(mge/module): fix named_tensors
GitOrigin-RevId: bb5aa1f41d
3 years ago
Megvii Engine Team
e5217db9f9
feat(traced_module): move traced_module out of the experimental folder
GitOrigin-RevId: 36c76b5277
3 years ago
Megvii Engine Team
442b4f6c26
test(traced_module): add some testcases for traced module
GitOrigin-RevId: 0d6bb20b2b
4 years ago
Megvii Engine Team
b2944559a8
fix(imperative/module): remove ``__getattribute__`` method in module
GitOrigin-RevId: 5ac525f010
4 years ago
Megvii Engine Team
1bf18252c4
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
4 years ago
Megvii Engine Team
c992afa960
test(mge/module): add module reset attribute test
GitOrigin-RevId: 6c9adc4a70
4 years ago
Megvii Engine Team
5a38ad3920
feat(mge/utils): add get/set_expand_structure to deal with complex key
GitOrigin-RevId: 4d1b952068
4 years ago
Megvii Engine Team
2763846168
fix(mge/module): support list/dict/tuple in module __repr__
GitOrigin-RevId: b70193fd79
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
8fad00a158
feat(mge/functional/nn): add conv1d padding
GitOrigin-RevId: 1bbfd36b96
4 years ago
Megvii Engine Team
20e93630f2
feat(mge): rename batch_norm2d -> batch_norm
GitOrigin-RevId: 253e8564ea
4 years ago
Megvii Engine Team
6ce796aec4
test(mge): replace `assertTensorClose` with `np.testing.assert_allclose`
GitOrigin-RevId: 2bf8c42cfd
4 years ago
Megvii Engine Team
477820fe5d
refactor(mge): apply python-module changes
GitOrigin-RevId: 2c769864c3
4 years ago
Megvii Engine Team
66f2dbd753
feat(mge/imperative): add module __repr__ for imperative
GitOrigin-RevId: ac13cc4659
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
0067dcf068
test(mge/imperative): add module test to imperative
GitOrigin-RevId: e690bc42b0
4 years ago