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 d8ea01680c docs(mge): update AvgPool2d & functional.nn.max_pool2d docstring 2 years ago
..
amp fix(imperative/amp): add is_scalar check in elemwise and concat 3 years ago
autodiff fix(imperative/amp): fix format transformation for symbol trans 3 years ago
core feat(gopt): support nchw44 global pooling with fuse_grain 2 years ago
data docs(dataloader): update dataloader docstring 2 years ago
distributed fix(mm): fix mm error when use sync 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): update AvgPool2d & functional.nn.max_pool2d docstring 2 years ago
hub chore(license): remove all license in file header 3 years ago
jit feat(serialization): add no_change_graph and version param whem dump model 3 years ago
module docs(mge): update AvgPool2d & functional.nn.max_pool2d docstring 2 years ago
optimizer fix(imperative): fix inplace operation of optim 2 years ago
quantization chore(license): remove all license in file header 3 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 feat(gopt): support nchw44 global pooling with fuse_grain 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(license): remove all license in file header 3 years ago
serialization.py chore(license): remove all license in file header 3 years ago
tensor.py feat(imperative/amp): add set_format 3 years ago