[发明专利]基于容器编排工具的智能训练装置

权利要求书

1.一种基于容器编排工具的智能训练装置，其特征在于，包括：

在容器编排工具模块基础上，集成API交互模块、数据处理模块和镜像库模块；

用户通过API交互模块输入训练参数，所述训练参数经所述容器编排工具模块转换和数据处理模块分析、优化处理后，由容器编排工具模块拉取镜像库模块的应用镜像并且启动容器进行训练。

2.按照权利要求1所述的基于容器编排工具的智能训练装置，其特征在于，还包括：

用户只需通过API交互模块输入训练参数并且通过API交互模块得到以图形化界面显示的训练结果，训练过程由数据处理模块、容器编排工具模块和镜像库模块完成。

3.按照权利要求1或2基于容器编排工具的智能训练装置，其特征在于，所述训练参数至少包括：

选择原始数据集；

选择网络结构；

键入求解参数；

选择资源类型及训练样本。

4.按照权利要求1或2所述的基于容器编排工具的智能训练装置，其特征在于，所述容器编排工具模块至少包含：

数据集转换模块，用以转换数据集；

优化处理模块，用以优化所述训练参数和推荐模型并配置参数，所述参数至少包括：display和snapshoot；

拉取应用镜像模块，用以拉取镜像库模块的应用镜像。

5.按照权利要求1或2所述的基于容器编排工具的智能训练装置，其特征在于，所述数据处理模块至少包含：

优化处理模块，用以对所述训练参数和所述训练结果进行解析和优化处理；

生成配置文件模块，用以根据所述训练参数和所述转换数据集生成配置文件供所述训练所用。

6.按照权利要求5所述的基于容器编排工具的智能训练装置，其特征在于，所述配置文件包含求解文件和网络文件，其中：

所述求解文件，至少包含参数：

base lr、lr policy、weight decay、momentum；

所述网络文件，至少包含：

池化层、数据层、激活层、准确率层。