模型转换方法、装置、电子设备和可读存储介质制造方法及图纸

本申请提出了一种模型转换方法、装置、电子设备和可读存储介质，其中，模型转换方法，包括：查找第一模型中的目标卷积算子；基于目标卷积算子的算子结构为预设结构，构建目标算子组，目标算子组的输出结果与目标卷积算子的输出结果相匹配；将第一模型中的目标卷积算子，替换为目标算子组，生成目标模型。生成目标模型。生成目标模型。

【技术实现步骤摘要】
模型转换方法、装置、电子设备和可读存储介质


[0001]本申请属于模型处理
，具体而言，涉及一种模型转换方法、装置、电子设备和可读存储介质。

技术介绍

[0002]通用推理框架对模型进行计算图优化的主要目的是加速前向推理，现有的计算图优化方案中加速推理是通过减少计算节点数量来实现的，减少计算节点数量的方法主要有算子融合和无用节点去除。
[0003]相关技术中，训练得到的模型中存在部分卷积节点使用不合理的情况，导致整个模型的推理速度较慢，通过算子融合和无用节点去除无法解决该问题。

技术实现思路

[0004]本申请旨在解决现有技术或相关技术中存在的技术问题之一。
[0005]为此，本申请的第一方面提出了一种模型转换方法。
[0006]本申请的第二方面提出了一种模型转换装置。
[0007]本申请的第三方面提出了一种模型转换装置。
[0008]本申请的第四方面提出了一种计算机程序产品。
[0009]本申请的第五方面提出了一种可读存储介质。
[0010]本申请的第六方面提出了一本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种模型转换方法，其特征在于，包括：查找第一模型中的目标卷积算子；基于所述目标卷积算子的算子结构为预设结构，构建目标算子组，所述目标算子组的输出结果与所述目标卷积算子的输出结果相匹配；将所述第一模型中的目标卷积算子，替换为所述目标算子组，生成目标模型。2.根据权利要求1所述的模型转换方法，其特征在于，所述基于所述目标卷积算子的算子结构为预设结构，构建目标算子组之前，还包括：读取所述目标卷积算子的卷积核的尺寸信息；在所述尺寸信息满足预设条件，且所述目标卷积算子的输入通道的数量为预设数量，确定所述目标卷积算子的所述卷积为预设结构。3.根据权利要求2所述的模型转换方法，其特征在于，所述尺寸信息包括数据宽度和数据高度；所述尺寸信息满足预设条件包括：所述数据宽度为1，且所述数据高度为n，n为所述目标卷积算子的输入数据的高度；或所述数据宽度为m，且所述数据高度为1，m为所述目标卷积算子的输入数据的宽度。4.根据权利要求1至3中任一项所述的模型转换方法，其特征在于，所述基于所述目标卷积算子的算子结构为预设结构，构建目标算子组，包括：获取所述目标卷积算子的数据计算信息，以及数据维度信息，所述数据维度信息为所述目标卷积算子输入数据和输出数据的维度信息；根据所述数据计算信息，构建目标等效算子；根据所述数据维度信息，构建重塑算子；根据所述重塑算子和所述目标等效算子，构建所述目标算子组。5.根据权利要求4所述的模型转换方法，其特征在于，所述根据所述数据计算信息，构建目标等效算子，包括：获取全连接算子；将所述数据计算信息，配置到所述全连接算子中，得到所述目标等效算子；其中，所述数据计算信息包括乘离率信息和权重信息。6.根据权利要求4所述的模型转换方法，其特征在于，所述重塑算子包括：输入重塑算子和输出重塑算子，所述数据维度信息包括数据输入维度和数据输出维度；所述根据所述数据维度信息，构建重塑算子，包括：根据所述数据输入维度，配置所述输入重塑算子；根据所述数据输出维度，配置所述输出重塑算子。7.根据权利要求6所述的模型转换方法，其特征在于，所述根据所述重塑算子和所述目标等效算子，构建所述目标算...

【专利技术属性】
技术研发人员：唐剑，夏立超，丁维浩，赵东宇，张法朝，奉飞飞，
申请(专利权)人：美的集团股份有限公司，
类型：发明
国别省市：