MegEngine

25 MB

Tree: 8d1f302590

Author	SHA1	Message	Date
Megvii Engine Team	8d1f302590	fix(mge): fix batch norm dump GitOrigin-RevId: `eb739437ef`	4 years ago
Megvii Engine Team	7d8a7e3efe	feat(mge): do not export F.loss.* GitOrigin-RevId: `a3ce4d3d76`	4 years ago
Megvii Engine Team	5b2f0129cb	feat(mge): rename cross_entropy_with_softmax -> cross_entropy GitOrigin-RevId: `9435c3260a`	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	66b6daf777	test(mge/optimizer): fix test for new optimizer api GitOrigin-RevId: `482ee62652`	4 years ago
Megvii Engine Team	fc3eca845b	feat(mge/imperative/jit): add dump input shape and xornet example GitOrigin-RevId: `5e2acd4052`	4 years ago