[发明专利]残差剪枝方法、装置、电子设备和可读存储介质

说明书

本申请提出了一种残差剪枝方法、装置、电子设备和可读存储介质，其中，残差剪枝方法，包括：查找第一模型对应的计算图中的目标节点，目标节点与第一模型中的残差连接结构相对应；基于目标节点遍历计算图，确定计算图中的目标卷积层；对目标卷积层进行剪枝处理。

技术领域

本申请属于模型处理技术领域，具体而言，涉及一种残差剪枝方法、装置、电子设备和可读存储介质。

背景技术

通道剪枝技术旨在压缩模型中的冗余权重，在缩小模型体积的同时，优化模型运行时的推理速度。

在相关技术中，自动化剪枝方案均需要人工手动输入所需剪枝处理的卷积层在模型中对应的名称，导致对模型自动化剪枝处理效率低下。

发明内容

本申请旨在解决现有技术或相关技术中存在的技术问题之一。

为此，本申请的第一方面提出了一种残差剪枝方法。

本申请的第二方面提出了一种残差剪枝装置。

本申请的第三方面提出了一种残差剪枝装置。

本申请的第四方面提出了一种计算机程序产品。

本申请的第五方面提出了一种可读存储介质。

本申请的第六方面提出了一种电子设备。

有鉴于此，根据本申请的第一方面提出一种残差剪枝方法，包括：查找第一模型对应的计算图中的目标节点，目标节点与第一模型中的残差连接结构相对应；基于目标节点遍历计算图，确定计算图中的目标卷积层；对目标卷积层进行剪枝处理。

根据本申请第二方面提出了一种残差剪枝装置，包括：查找模块，用于查找第一模型对应的计算图中的目标节点，目标节点与第一模型中的残差连接结构相对应；确定模块，用于基于目标节点遍历计算图，确定计算图中的目标卷积层；处理模块，用于对目标卷积层进行剪枝处理。

根据本申请第三方面提出了一种残差剪枝装置，包括：存储器，存储器中存储有程序或指令；处理器，处理器执行存储在存储器中的程序或指令以实现如第一方面中任一技术方案中的残差剪枝方法的步骤，因而具有上述第一方面中任一技术方案中的残差剪枝方法的全部有益技术效果，在此不再做过多赘述。

根据本申请第四方面提出了一种计算机程序产品，计算机程序产品被处理器执行时实现如第一方面中任一技术方案中的残差剪枝方法的步骤，因而具有上述第一方面中任一技术方案中的残差剪枝方法的全部有益技术效果，在此不再做过多赘述。

根据本申请第五方面提出了一种可读存储介质，可读存储介质上存储有程序或指令，程序或指令被处理器执行时实现如上述第一方面中任一技术方案中的残差剪枝方法的步骤。因而具有上述第一方面中任一技术方案中的残差剪枝方法的全部有益技术效果，在此不再做过多赘述。

根据本申请第六方面提出了一种电子设备，包括：如上述第二方面或第三方面中限定的残差剪枝装置，和/或上述第四方面中限定的可读存储介质，因而具有上述第二方面或第三方面中残差剪枝装置，和/或上述第四方面中限定的可读存储介质的全部有益技术效果，在此不再做过多赘述。

本申请的技术方案中，通过将第一模型转换为计算图结构，并基于计算图结构查找到目标节点，再通过目标节点查找到所需进行剪枝处理的目标卷积层，自动对目标卷积层进行定位，并对定位得到的目标卷积层进行自动剪枝处理，实现了自动化获取模型中的残差连接结构的卷积层连接关系，无需用户手动输入这些卷积层在模型中对应的名称，保证对模型剪枝处理准确性的同时，提高模型剪枝处理的效率。

本申请的附加方面和优点将在下面的描述部分中变得明显，或通过本申请的实践了解到。

附图说明

本申请的上述和/或附加的方面和优点从结合下面附图对实施例的描述中将变得明显和容易理解，其中：