Cambricon ModelZoo是由寒武纪开发和维护的模型合集,提供人工智能各个应用领域(包括并不限于CV/NL/语音/推荐等)中经典和SOTA模型的训练和推理应用示例。
本仓库旨在基于寒武纪的硬件产品和软件SDK,展示最佳的编程实践,以达成模型的快速移植和最优性能,开发者可以从modelzoo的源码中得到研究和二次开发的参考。
本仓库在持续更新中,如果您有任何需求,可以在modelzoo/issue或者寒武纪开发者论坛给我们提issue,我们会尽快评估和处理,也欢迎在寒武纪开发者社区浏览我们的更多内容。
Cambricon ModelZoo仓库包含5个子仓库,分别是 magicmind_cloud_modelzoo,magicmind_edge_modelzoo , cambricon_pytorch_modelzoo,cambricon_tensorflow_modelzoo 和 paddle_mlu_modelzoo 您可以直接克隆本仓库,或者分别克隆需要的子仓库,并根据子仓库的README指引进行评测和开发。
子仓库 | 说明 |
---|---|
magicmind_cloud_modelzoo | 基于寒武纪自研推理引擎magicmind在云端产品的模型推理集合 |
magicmind_edge_modelzoo | 基于寒武纪自研推理引擎magicmind在边端产品的模型推理集合 |
cambricon_pytorch_modelzoo | 基于Cambricon PyTorch的模型训练/推理集合 |
cambricon_tensorflow_modelzoo | 基于Cambricon Tensorflow的模型训练/推理集合 |
paddle_mlu_modelzoo | 百度paddle运行在MLU上的模型集合 |
TBD
Cambricon ModelZoo的License具体内容请参见LICENSE文件。
Cambricon ModelZoo仅提供公共数据集以及预训练模型的下载链接,公共数据集及预训练模型并不属于ModelZoo, ModelZoo也不对其质量或维护承担责任。请您在使用公共数据集和预训练模型的过程中,确保符合其对应的使用许可。
如果您不希望您的数据集或模型公布在ModelZoo上,或者您希望更新ModelZoo中属于您的数据集或模型,请您通过Gitee中提交issue,您也可以联系[email protected]告知我们。