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