Skip to content

Latest commit

 

History

History
37 lines (26 loc) · 2.94 KB

README.md

File metadata and controls

37 lines (26 loc) · 2.94 KB

用户指南

1. 安装和部署

Vega运行环境为:Ubuntu 18.04、CUDA 10.0、Python3.7,并支持集群部署,详细的安装和部署指导,可参考 安装指导集群部署指导

2. 使用指导

Vega提供了丰富的AutoML任务,用户可根据下表选择合适的算法:

任务分类参考算法
图像分类网络架构搜索CARSNAGO、BackboneNas、DartsCNN、GDAS、EfficientNet
超参优化ASHA、BOHB、BOSS、BO、TPE、Random、Random-Pareto
数据增广PBA
模型压缩模型剪枝Prune-EA
模型量化Quant-EA
图像超分辨率网络架构搜索SR-EAESR-EA
数据增广CycleSR
图像语义分割网络架构搜索Adelaide-EA
物体检测网络架构搜索SP-NAS
车道线检测网络架构搜索Auto-Lane
推荐搜索特征选择AutoFIS
特征交互建模AutoGroup

Vega针对以上每个算法提供了示例,通过尝试运行示例,可快速掌握各个算法。具体运行示例的方法,可参考 示例参考

通过运行示例,会发现一般Vega算法的输入是数据集和配置信息,输出为网络架构描述、模型训练超参、或模型。同时可根据需要将不同的算法和fully train步骤组合成一个pipeline,实现端到端的AutoML流程,输入数据,即可得到所需的模型。

如上描述Pipeline的详细配置信息,可参考 配置指导

3. 模型端侧评估

Vega还提供了端侧模型评估的能力,支持的端侧硬件有Davinci推理芯片(ATLAS 200 DK、ATLAS 300产品和开发板环境Evb)和手机,支持在 Bolt 部署评估。

用户可参考 评估服务安装和配置指导 安装和配置评估服务,用以在架构搜索过程中实时评估搜索到的模型,得到适用于该终端设备的模型。