Megvii Engine Team
c2e9860feb
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
3 years ago
Megvii Engine Team
e6caa9ff89
feat(opr): add bn backward for inference mode
GitOrigin-RevId: bb643cb62f
4 years ago
Megvii Engine Team
070c811732
fix(imperative): remove convert_inputs
GitOrigin-RevId: a3c43db746
4 years ago
Megvii Engine Team
e82fa4ec23
fix(gopt): using new_inp for build_chain in DelayBroadcast pass
GitOrigin-RevId: efc6377197
4 years ago
Megvii Engine Team
27a9326c66
fix(mge/module): fix frozen batch norm
GitOrigin-RevId: 143d468a37
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
7cd846a570
fix(mge/batch_norm): fix batch_norm check when trace(symbolic=True)
GitOrigin-RevId: 2032eb5f7d
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
e283663a02
fix(mge/imperative): update tests to new optimizer api
GitOrigin-RevId: 3d06e3db3c
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