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 4755400eda refactor(mge/quantization): add `narrow_range` to control quant dtype's lower bound 5 years ago
..
pytorch fix(mge/module): fix torch subgraph under jit.trace with symbolic=False 5 years ago
qat refactor(mge/quantization): add `narrow_range` to control quant dtype's lower bound 5 years ago
quantized feat(mge/module): add conv and conv_relu quantization module 5 years ago
__init__.py feat(mge/module): add conv and conv_relu quantization module 5 years ago
activation.py docs(mge/module): refine the docstring of several apis 5 years ago
batchnorm.py feat(mge/module): add sync bn 5 years ago
concat.py refactor(mge/quantization): split `QATModule` and refactor convert api 5 years ago
conv.py feat(mge/module): add conv and conv_relu quantization module 5 years ago
conv_bn.py feat(mge/module): add conv and conv_relu quantization module 5 years ago
dropout.py docs(mge/module): refine the docstring of several apis 5 years ago
elemwise.py refactor(mge/quantization): split `QATModule` and refactor convert api 5 years ago
embedding.py test(mge): move `NUMBER` config to pytest.init 5 years ago
identity.py docs(mge/module): refine the docstring of several apis 5 years ago
init.py docs(mge/module): refine the docstring of several apis 5 years ago
linear.py refactor(mge/quantization): split `QATModule` and refactor convert api 5 years ago
module.py fix(mge/module): fix redundant recursion in `train()` 5 years ago
parampack.py feat(mge/parampack): add user-defined key to pack params 5 years ago
pooling.py MegEngine: Initial commit of MegEngine. 5 years ago
quant_dequant.py refactor(mge/quantization): split `QATModule` and refactor convert api 5 years ago
sequential.py feat(mge/quantization): add quantization interface 5 years ago

MegEngine 安装包中集成了使用 GPU 运行代码所需的 CUDA 环境,不用区分 CPU 和 GPU 版。 如果想要运行 GPU 程序,请确保机器本身配有 GPU 硬件设备并安装好驱动。 如果你想体验在云端 GPU 算力平台进行深度学习开发的感觉,欢迎访问 MegStudio 平台

C++ Cuda Python C SVG other