You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
Megvii Engine Team b50947d2e2 docs(mge/module): add Module.state_dict docstring 2 years ago
..
amp fix(imperative/amp): add is_scalar check in elemwise and concat 3 years ago
autodiff chore(mge/misc): rename APIs that should not be public 2 years ago
core feat(gopt): support nchw44 global pooling with fuse_grain 2 years ago
data chore(mge/misc): rename APIs that should not be public 2 years ago
distributed chore(mge/misc): rename APIs that should not be public 2 years ago
dtr chore(license): remove all license in file header 3 years ago
experimental chore(license): remove all license in file header 3 years ago
functional docs(mge/functional): update copy example 2 years ago
hub chore(mge/misc): rename APIs that should not be public 2 years ago
jit chore(mge/misc): rename APIs that should not be public 2 years ago
module docs(mge/module): add Module.state_dict docstring 2 years ago
optimizer fix(imperative): fix inplace operation of optim 2 years ago
quantization fix(mge/quant): call is_distributed func correctly 2 years ago
random chore(license): remove all license in file header 3 years ago
tools feat(imperative): load_nerwork_and_run enable weight preprocess 2 years ago
traced_module fix(traced_module): fix trace_module function may raise error in finally scope 3 years ago
utils fix(mge): update format 2 years ago
__init__.py fix(mgb): change the check method of cuda sm code 2 years ago
_multistream.py perf(mge): improve cross stream memory borrowing 3 years ago
config.py fix(mge): control compute_mode by context 3 years ago
conftest.py ci(pytest): fixture doctest namespace in pytest config 3 years ago
device.py fix(mge/device): add missed API to __all__ scope 2 years ago
logger.py chore(mge/misc): rename APIs that should not be public 2 years ago
serialization.py chore(mge/misc): rename APIs that should not be public 2 years ago
tensor.py fix(mge/tensor): modifiy tensor print suppress options 2 years ago