Megvii Engine Team
0075e6acb0
refactor(mge): refactor cross_entropy
GitOrigin-RevId: 1fac5b5b14
4 years ago
Megvii Engine Team
37e56f4b04
fix(mge/functional): fix clip under trace(symbolic=True)
GitOrigin-RevId: 5b6f537327
4 years ago
Megvii Engine Team
af349d6102
fix(mge/functional): fix op mismatch when tracing NMSKeep
GitOrigin-RevId: e8f2cbb755
4 years ago
Megvii Engine Team
d502e79f85
feat(mge): make F.norm numpy compatible
GitOrigin-RevId: 4e5236171c
4 years ago
Megvii Engine Team
d31a4fff73
refactor(mge): use logsumexp in cross_entropy
GitOrigin-RevId: 4a14aabb94
4 years ago
Megvii Engine Team
20e93630f2
feat(mge): rename batch_norm2d -> batch_norm
GitOrigin-RevId: 253e8564ea
4 years ago
Megvii Engine Team
aa62672629
feat(mge): make F.eye numpy compatible
GitOrigin-RevId: fee32537b4
4 years ago
Megvii Engine Team
7589bce454
feat(mge): do not export F.nn.sync_batch_norm
GitOrigin-RevId: aa05e3aac4
4 years ago
Megvii Engine Team
d4e2d27213
feat(mge): do not export F.nn.conv_bias_activation
GitOrigin-RevId: e536f87fad
4 years ago
Megvii Engine Team
e64760fe2b
docs(mge): warp_perspective, gather & scatter
GitOrigin-RevId: f2436e8501
4 years ago
Megvii Engine Team
7d8a7e3efe
feat(mge): do not export F.loss.*
GitOrigin-RevId: a3ce4d3d76
4 years ago
Megvii Engine Team
480375fbe3
feat(mge): rename accuracy -> topk_accuracy
GitOrigin-RevId: b6ae007231
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
cf0507e1ba
feat(mge): remove F.identity
GitOrigin-RevId: 858be627ac
4 years ago
Megvii Engine Team
09241a1ff7
feat(mge): remove param_pack_* from functional
GitOrigin-RevId: a5fe25be8c
4 years ago
Megvii Engine Team
0d4568d6f8
feat(mge): rename broadcast -> broadcast_to
GitOrigin-RevId: 82f46ad2c2
4 years ago
Megvii Engine Team
f551d44e37
feat(mge): rename arange paramter end -> stop
GitOrigin-RevId: ea9064e351
4 years ago
Megvii Engine Team
9c5ea68992
feat(mge): remove dimshuffle
GitOrigin-RevId: 9e327ee3af
4 years ago
Megvii Engine Team
1e2117f66d
feat(mge): remove add_axis and remove_axis
GitOrigin-RevId: 59611d43f9
4 years ago
Megvii Engine Team
5b7ae26849
feat(mge): do not export F.nn.roi_pooling
GitOrigin-RevId: 8a07dd1da4
4 years ago
Megvii Engine Team
d22d1676e7
feat(mge): do not export F.nn.roi_align
GitOrigin-RevId: c2ac7bab32
4 years ago
Megvii Engine Team
9c5611345f
feat(mge): do not export F.nn.nms
GitOrigin-RevId: 7a0a328c37
4 years ago
Megvii Engine Team
56d624f5be
feat(mge): rename log_softmax -> logsoftmax
GitOrigin-RevId: 40ccfea5a7
4 years ago
Megvii Engine Team
164d9cdf1f
feat(mge): do not export F.nn.linear
GitOrigin-RevId: 589964729d
4 years ago
Megvii Engine Team
e42679b517
feat(mge): do not export F.nn.interpolate
GitOrigin-RevId: 031c6555c0
4 years ago
Megvii Engine Team
fd802e978f
feat(mge): do not export F.nn.embedding
GitOrigin-RevId: eb90506535
4 years ago
Megvii Engine Team
a640eeff62
feat(mge): remove batched_nms
GitOrigin-RevId: 01f9ee137c
4 years ago
Megvii Engine Team
205a39b07d
feat(mge): remove add_update
GitOrigin-RevId: 3068593eeb
4 years ago
Megvii Engine Team
67d33303ad
feat(mge): remove fast_tanh
GitOrigin-RevId: 7ad5f7ecce
4 years ago
Megvii Engine Team
78d2189441
feat(mge): rename comparison operators to their full name
GitOrigin-RevId: e503b038a1
4 years ago
Megvii Engine Team
e034874ae7
feat(mge): rename clamp -> clip
GitOrigin-RevId: 2b594a611e
4 years ago
Megvii Engine Team
e0da74852e
feat(opr): add confidential operator
GitOrigin-RevId: 53c2d4bc45
4 years ago
Megvii Engine Team
20cdd11d4d
docs(mge): fix docstring of cgtools
GitOrigin-RevId: ee8c93b9f7
4 years ago
Megvii Engine Team
2bd84d6789
feat(mge): add adaptive pooling python wrapper
GitOrigin-RevId: 789f1511ec
4 years ago
Megvii Engine Team
0e3037105c
feat(mge/module): add group_local_conv
GitOrigin-RevId: 0503377bc5
4 years ago
Megvii Engine Team
026af62042
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
4 years ago
Megvii Engine Team
2070aaae09
fix(mge/functional): fix duplicate `transpose` in `__all__` and add `dimshuffle
GitOrigin-RevId: 2967088233
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
a398d4b5a5
chore(mge/functional): remove duplicated code
GitOrigin-RevId: f9efea46cb
4 years ago
Megvii Engine Team
06041f8a7e
fix(mge/functional): fix trace warp_perspective
GitOrigin-RevId: 2071bb63a8
4 years ago
Megvii Engine Team
f4860b9345
fix(mge/functional): fix trace topk
GitOrigin-RevId: c88ca8219b
4 years ago
Megvii Engine Team
1a24fb29c1
perf(mge/allreduce): put allreduce on another cuda stream
GitOrigin-RevId: 2e778dfa04
4 years ago
Megvii Engine Team
b970c6daf1
fix(functional): hide format interface and skip test_conv_bias
GitOrigin-RevId: 7de75bf084
4 years ago
Megvii Engine Team
aac8de554e
fix(mge/parampacksplit): fix parampacksplit refcnt error
GitOrigin-RevId: c964465596
4 years ago
Megvii Engine Team
715009b564
refactor(mge/api): remove external, dropout fix
GitOrigin-RevId: 5e6ff1a372
4 years ago
Megvii Engine Team
70bc746c20
fix(mge): fix bug of identity()
GitOrigin-RevId: 4bfd3cafb5
4 years ago
Megvii Engine Team
a8c75ee530
refactor(mge/functional): make docs better
GitOrigin-RevId: acb287d48b
4 years ago
Megvii Engine Team
6c1dbd40d9
docs(mge): fix doctest
GitOrigin-RevId: 131fed8733
4 years ago
Megvii Engine Team
39d4328309
fix(mge/functional): functional api fixes
GitOrigin-RevId: fa206c4ff6
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