Megvii Engine Team
99fae95e02
feat(mge/parampack): add user-defined key to pack params
GitOrigin-RevId: 7d51dcae23
5 years ago
Megvii Engine Team
9a42c63641
fix(mge/module): add the more warnings in load_state_dict
In the before, the information of miss matching and unused operators won't be printed in the non-trict mode, this commit add the information.
GitOrigin-RevId: b2543eb832
5 years ago
Megvii Engine Team
779ed45d74
docs(mge/parampack): refine parampack doc
GitOrigin-RevId: 1cff83c308
5 years ago
Megvii Engine Team
7ed98dd50a
feat(mge/functional): add conv_transpose2d support
GitOrigin-RevId: 9ad87a4ea9
5 years ago
Megvii Engine Team
f077a5292c
fix(mge/param_pack): release old parameters
GitOrigin-RevId: 40a1f044e9
5 years ago
Megvii Engine Team
48822557b8
docs(mge/bn): fix docs and tests of batchnorm
GitOrigin-RevId: 8a96aa5fc2
5 years ago
Megvii Engine Team
2c4d1afe14
test(mge): move `NUMBER` config to pytest.init
GitOrigin-RevId: 1d82209c40
5 years ago
Megvii Engine Team
e1b2d31d36
test(mge): skip some doctests having different results on gpu
GitOrigin-RevId: 66e5db5c89
5 years ago
Megvii Engine Team
62b0c6cd27
style(mge): apply format.sh
GitOrigin-RevId: a900b1bb6e
5 years ago
Megvii Engine Team
ea00b57d93
refactor(mge/module): use mge rng to sample from uniform/normal distribution
GitOrigin-RevId: 6ec8f99af5
5 years ago
Megvii Engine Team
a1597cfc46
docs(mge/param_pack): add description of param pack and add param pack in example run_dist
GitOrigin-RevId: 1c900e159d
5 years ago
Megvii Engine Team
80d3a06f9e
fix(mge/param_pack): fix param pack with just one param
GitOrigin-RevId: 0d28a12e59
5 years ago
Megvii Engine Team
a744b3cb8a
feat(mge/module):add param pack
GitOrigin-RevId: 9cf1dbe44d
5 years ago
Megvii Engine Team
65432d3b93
fix(mge/module): fix torch subgraph under jit.trace with symbolic=False
GitOrigin-RevId: a208ba79d9
5 years ago
Megvii Engine Team
f91881ffdc
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
5 years ago