48 Commits (a601fda89fea416f85d72280018a06a0ecb82492)

Author SHA1 Message Date
  Megvii Engine Team 3f2770fe03 feat(mge/module): add quantize dtype load support for module load_state_dict 5 years ago
  Megvii Engine Team c8a9094bb3 fix(mge/data/dataloader): add refcount in _PlasmaStoreManager 5 years ago
  Megvii Engine Team fda9599a84 feat(mge/quant): add TQT quant method 5 years ago
  Megvii Engine Team 5c2323529d test(mge/quantization): add `quantize_disabled` related test 5 years ago
  Megvii Engine Team 08ac685e3a feat(mge/functional): add logsumexp 5 years ago
  Megvii Engine Team c7e6c658fd refactor(mge/distribute): use is_root (and rank) in stead of rank and root at collective comm 5 years ago
  Megvii Engine Team 35bc0e1f60 fix(mge/function): do not deeply copy saved tensor in Function 5 years ago
  Megvii Engine Team d2f5874a52 fix(mge/module): fix non-str key error of dict in module 5 years ago
  Megvii Engine Team e6820b91e8 feat(mge/module): add conv and conv_relu quantization module 5 years ago
  Megvii Engine Team 40412e2688 feat(mge/module): add sync bn 5 years ago
  Megvii Engine Team caf1fac251 refactor(mge/quantization): split `QATModule` and refactor convert api 5 years ago
  Megvii Engine Team bb872965e3 feat(mge/module): add local conv2d module 5 years ago
  Megvii Engine Team f7d8b51648 feat(mge/functional): add smooth l1 loss 5 years ago
  Megvii Engine Team 3c49d1d324 feat(mge/functional): add hinge loss 5 years ago
  Megvii Engine Team 7aa7a09b6b feat(mge/optimizer): add optimizer adadelta 5 years ago
  Megvii Engine Team 5eca4da323 feat(mge/functional): add softplus function 5 years ago
  Megvii Engine Team 90107b6dda feat(mge/cgtools): add network vistior interface with optional pruning 5 years ago
  Megvii Engine Team 980ebf2c72 feat(mge/module): add fused conv_bn qat approximate version 5 years ago
  Megvii Engine Team 3adad4852d feat(mge/optimizer): add optimizer adagrad 5 years ago
  Megvii Engine Team 8be78b114e test(mge/optimizer): refactor the unittest of optimizer 5 years ago
  Megvii Engine Team b3b14fdfe1 fix(mge/jit): fix add_update semantic 5 years ago
  Megvii Engine Team b2ed2630f7 Revert "fix(mge/functional): rm useless arguments" 5 years ago
  Megvii Engine Team d23fec06cb fix(mge/functional): rm useless arguments 5 years ago
  Megvii Engine Team 4500faf1ea fix(python_module/mge): one_hot: no default value for num_classes 5 years ago
  Megvii Engine Team 8c110c3942 feat(mge/quantization): add quantization interface 5 years ago
  Megvii Engine Team 814a427810 feat(mge/module): add `with_parent` argument in `_flatten` 5 years ago
  Megvii Engine Team d3730036ea test(mge/dist): add unit tests for megengine.distributed.utils 5 years ago
  Megvii Engine Team 80d9e56b38 fix(mge/tensor): where support x or y is empty 5 years ago
  Megvii Engine Team 4c5d01fa34 refactor(mge/examples): add example to use modified sublinear API 5 years ago
  Megvii Engine Team 396653df5f feat(mge/api): expose sublinear related parameters at mge api level 5 years ago
  Megvii Engine Team dd962a4700 fix(mge/module): add negative slope attribute for LeakyReLU module 5 years ago
  Megvii Engine Team 5507a29bd3 feat(mge/functional): add normalize opr 5 years ago
  Megvii Engine Team 000663c30c feat(mge/functional): add isnan and isinf oprs 5 years ago
  Megvii Engine Team 9b41321926 fix(mge/functional): support negative axis in math.py 5 years ago
  Megvii Engine Team 207527d108 fix(mge/clamp): fix `F.clamp` 5 years ago
  Megvii Engine Team bb92ee26e3 fix(mge/functional): refine doc and add test for linspace 5 years ago
  Megvii Engine Team ce30d045ee feat(mge/functional): add arange in tensor.py 5 years ago
  Megvii Engine Team fc4dedc0bc fix(mge/random): add lower bound and higher bound for uniform sampling 5 years ago
  Megvii Engine Team 712b87c8c1 feat(mgb/core): add comp node for cambricon 5 years ago
  Megvii Engine Team d74a4ee931 fix(mge/build): use consistent genfiles/megbrain_build_config.h 5 years ago
  Megvii Engine Team e8494f729b fix(unittest): remove the torch dependency in test_conv 5 years ago
  Megvii Engine Team 7ed98dd50a feat(mge/functional): add conv_transpose2d support 5 years ago
  Megvii Engine Team 044695aa2a feat(mgb/plugin): add profiler for dynamic graph 5 years ago
  Megvii Engine Team 62b0c6cd27 style(mge): apply format.sh 5 years ago
  Megvii Engine Team dd41e2e47a fix(mge/test):fix integration/manual tests 5 years ago
  Megvii Engine Team 8d2bbf7383 fix(mge/functional): support scalar inputs in elemwise functions 5 years ago
  Megvii Engine Team 65432d3b93 fix(mge/module): fix torch subgraph under jit.trace with symbolic=False 5 years ago
  Megvii Engine Team f91881ffdc MegEngine: Initial commit of MegEngine. 5 years ago