Megvii Engine Team
ee4ea7fdc8
test(distributed/test): make distributed test more stronger
GitOrigin-RevId: 085fd1dcfd
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
442058aead
feat(mge/quantize): add batch matmul activation module for inference
GitOrigin-RevId: f6dbd1f4c0
4 years ago
Megvii Engine Team
33e8879af4
feat(mge/quantization): support distributed qat
GitOrigin-RevId: c915c843b8
4 years ago
Megvii Engine Team
9cb3c07cd4
feat(mge/functional): add elemwise mode support string input
GitOrigin-RevId: 57be5cec7b
4 years ago
Megvii Engine Team
8fad00a158
feat(mge/functional/nn): add conv1d padding
GitOrigin-RevId: 1bbfd36b96
4 years ago
Megvii Engine Team
2627e1f7df
fix(mge/grad_manager): allow multiple calls of `release`
GitOrigin-RevId: 38ca4c78ff
4 years ago
Megvii Engine Team
dfc3935d06
fix(pytest/windows/impertive): fix windows wrong test skip
GitOrigin-RevId: ca0f04cfe7
4 years ago
Megvii Engine Team
9748aebeea
refactor(mge): tensor_shape -> symbolic_shape
GitOrigin-RevId: 366dc048bf
4 years ago
Megvii Engine Team
20e93630f2
feat(mge): rename batch_norm2d -> batch_norm
GitOrigin-RevId: 253e8564ea
4 years ago
Megvii Engine Team
2efba9a3a3
fix(mgb/test): use both rtol and atol for stable test result
GitOrigin-RevId: 82a1453e4a
4 years ago
Megvii Engine Team
0e3037105c
feat(mge/module): add group_local_conv
GitOrigin-RevId: 0503377bc5
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
bd7f885a2b
fix(mge/pytest): remove __init__ in py-test
GitOrigin-RevId: e16eae9cbc
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
2beb65b19d
fix(mge/module): tensor shape will not work when constructing numpy array
GitOrigin-RevId: 5a0d705970
4 years ago
Megvii Engine Team
a226f02e7a
fix(mge/imperative): fix syncbn in symbolic mode
GitOrigin-RevId: a9794318a7
4 years ago
Megvii Engine Team
0067dcf068
test(mge/imperative): add module test to imperative
GitOrigin-RevId: e690bc42b0
4 years ago