神经结构搜索模型的训练方法、装置及电子设备

专利申请号：CN202211447618.3

公 开 号：CN116341634A

发 明 人：张恒 

代 理 人：北京法胜知识产权代理有限公司黄海艳

代理机构：北京法胜知识产权代理有限公司

专利类型：发明专利

申 请 日：20230627

公 开 日：20221118

专利主分类号：G06N3/08

关 键 词：推理 神经网络 编译器 神经结构 硬件配置 搜索 参数组合 关联数据 存储介质 电子设备 结构输入 结构搜索 目标神经 加载 申请 引入 

摘      要：本申请公开了神经结构搜索模型的训练方法及装置、电子设备和存储介质，主要技术方案包括：将不同参数组合的神经网络元结构输入神经结构搜索模型，编译器模型包含多种编译器的硬件配置；获取神经网络元结构加载至编译器模型，确定不同的编译器硬件配置推理不同参数组合的神经网络元结构的推理时间，并获得神经网络元结构与推理时间的关联数据；根据神经网络元结构与推理时间的关联数据对神经结构搜索模型进行训练，以得到目标神经结构搜索模型。本申请在神经结构搜索模型中引入编译器模型，确定编译器模型中不同硬件配置编译器推理神经网络元结构的推理时间，更准确的获得神经网络元结构和推理速度(推理时间)之间的关系。