MegEngine模型
本仓库包含了采用实现的各种主流深度学习模型。
目录下提供了各种经典的图像分类,目标检测,图像分割以及自然语言模型的官方实现。每个模型同时提供了模型定义,推理以及训练的代码。
官方会一直维护下代码,保持适应MegEngine的最新API,提供最优的模型实现。同时,提供替代的学习文档,帮助新手学习如何在MegEngine下训练自己的模型。
概述
对于每个模型,我们提供了至少四个脚本文件:模型定义( model.py ),模型推理( inference.py ),模型训练( train.py ),模型测试( test.py )。
每个模型目录下都柏林对应有一个README ,介绍了模型的详细信息,并详细描述了训练和测试的流程。例如。
另外, official下定义的模型可以通过megengine.hub来直接加载,例如:
import megengine.hub
# 只
1