[发明专利]模型获取方法、电子设备及存储介质

说明书

本申请提供了一种模型获取方法、电子设备及存储介质，其中，模型获取方法，通过获取多种部署终端的硬件结构所对应的第一信息、及用于构建模型的多种网络结构单元的模型单元信息；当接收到适配指令时，根据所述适配指令将每一所述第一信息与至少一所述模型单元信息适配关联，得到适配信息；获取目标模型的模型需求信息，并根据所述模型需求信息获取目标部署终端的目标第一信息，所述目标部署终端用于部署所述目标模型；根据所述目标第一信息及所述适配信息，确定可用于构建所述目标模型的多组目标网络结构单元对应的目标模型单元信息；根据所述目标模型单元信息及所述模型需求构建初始模型，并从所述初始模型中筛选所述目标模型。

技术领域

本申请涉及人工智能技术领域，尤其涉及一种模型获取方法、电子设备 及存储介质。

背景技术

自动机器学习在模型构建、模型训练和评估的过程中通常利用模型搜索 方法，获取可以部署于对应终端设备，以实现模型结构和模型参数的自动优 化。基于模型所部署的终端设备存在多种类型，不同的终端设备所对应的硬 件类型不同，如，不同的终端设备对应的硬件厂商不同，硬件平台不同，芯 片运行IP不同，造成不同终端设备所适配的最优模型对应的模型结构、模型大 小各有不同。

现有的模型搜索方法是在搜索空间选择一些模型进行训练，并根据训练 结果获取目标模型，然而，通过该搜索方法所获取的目标模型可能无法充分 发挥终端设备硬件能效的情况，用户体验较差。

发明内容

本申请提供了一种模型获取方法、电子设备及存储介质，旨在获取与部 署终端最优的适配模型，以充分发挥部署终端的硬件能效，提升用户体验。

第一方面，本申请实施例提供了一种模型获取方法，包括：

获取多种部署终端的硬件结构所对应的第一信息、及用于构建模型的多 种网络结构单元的模型单元信息；

当接收到适配指令时，根据所述适配指令将每一所述第一信息与至少一 所述模型单元信息适配关联，得到适配信息；

获取目标模型的模型需求信息，并根据所述模型需求信息获取目标部署 终端的目标第一信息，所述目标部署终端用于部署所述目标模型；

根据所述目标第一信息及所述适配信息，确定可用于构建所述目标模型 的多组目标网络结构单元对应的目标模型单元信息；

根据所述目标模型单元信息及所述模型需求构建初始模型，并从所述初 始模型中筛选所述目标模型。

本实施方式中，通过将多种部署终端的硬件结构所对应的第一信息及该 硬件结构所适配的对应网络结构单元所对应的模型单元信息进行适配，从而 获取适配信息，在接收到用户的模型需求信息时，根据模型需求信息解析到 目标部署终端的硬件结构对应的目标第一信息，从而通过适配信息为该目标 第一信息适配对应的目标网络结构单元信息，从而根据目标网络结构单元信 息获取到与目标部署终端的硬件结构适配的目标网络结构单元，从而可以根 据目标网络结构单元构建初始模型，并从初始模型中筛选出符合预设需求的 目标模型。基于构建模型的目标网络结构单元与目标部署终端的硬件结构适 配，可以充分发挥出目标部署终端的硬件结构的硬件能效，从而确保最终所 获取的目标模型也可以充分发挥目标部署终端的硬件结构的硬件能效，进而 提升目标模型在目标部署终端的运行体验。

在一些实施方式中，所述获取多种部署终端的硬件结构所对应的第一信 息，包括：

确定多种部署终端的硬件结构对应的第二信息；

根据所述第二信息确定对应所述硬件结构的第一信息。

本实施方式中，第二信息包括但不限定于网络可以获取的硬件结构所公 开的信息，如，硬件结构的官方网址信息，通过网络所公开的对应部署终端 的硬件结构的信息综合筛选出目标部署终端的硬件结构所对应的第一信息， 从而确保所获取的第一信息的精准性，全面性。