Megvii Engine Team
7a9d1f57cd
fix(mge/module): fix module attribute update mistake
GitOrigin-RevId: 842a8fcb6e
4 years ago
Megvii Engine Team
aed681d325
feat(imperative/utils): optimize the naming rules
GitOrigin-RevId: 329bac640a
4 years ago
Megvii Engine Team
48a7d333c3
fix(mge/module): fix auto-naming error when there are containers in the module
GitOrigin-RevId: 6b8f043321
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
1d7dd00144
feat(mge/quantization): add QParams and QuantDtypeMeta for quantization data structure
GitOrigin-RevId: df3416fe13
4 years ago
Megvii Engine Team
a1ca50c923
feat(mge/quantization): add name for quantized module
GitOrigin-RevId: edefbec7b7
4 years ago
Megvii Engine Team
6fb19b664b
feat(imperative/src): name operators automatically when tracing
GitOrigin-RevId: ff8eb003c5
4 years ago
Megvii Engine Team
dbb3dd681f
refactor(profiler): integrate profiler into interpreter
GitOrigin-RevId: ccc984acbd
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
6c4841e807
fix(mge/quantization): `disable_fake_quant` does not work correctly
GitOrigin-RevId: 0c568d3335
4 years ago
Megvii Engine Team
7167fdbd49
feat(mge/module): add normalization module includes group_norm, instance_norm and layer_norm
GitOrigin-RevId: 9f253e3250
4 years ago
Megvii Engine Team
94dba16ff4
perf(mge/imperative): misc optimizations
GitOrigin-RevId: bbe7a10b00
4 years ago
Megvii Engine Team
82b0f67770
fix(mge/core): fix dtype promotion issue for quantized dtype
GitOrigin-RevId: 9d09c8fa6f
4 years ago
Megvii Engine Team
b309890c66
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
4 years ago
Megvii Engine Team
6701346398
docs(mge): fix some docstring format problem
GitOrigin-RevId: cbc5ab04b3
4 years ago
Megvii Engine Team
026af62042
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
4 years ago
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
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
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