模型剪枝方法、装置和计算机设备

    公开(公告)号:CN114819140B

    公开(公告)日:2025-05-13

    申请号:CN202210330396.0

    申请日:2022-03-31

    Abstract: 本说明书实施例公开了一种模型剪枝方法、装置和计算机设备。所述方法包括:根据剪枝参数,确定掩码信息;所述掩码信息用于指示目标模型中剪枝对象的有效状态;将样本输入至增添了掩码信息后的目标模型,得到目标模型的第一输出;根据第一输出,优化参数信息;所述参数信息包括目标模型的模型参数和剪枝参数;迭代执行以上步骤,直至满足结束条件;根据掩码信息,对剪枝对象进行剪枝处理。本说明书实施例可以对目标模型进行剪枝处理,以减少存储资源和计算资源的占用。

    模型剪枝方法、装置和计算机设备

    公开(公告)号:CN114819140A

    公开(公告)日:2022-07-29

    申请号:CN202210330396.0

    申请日:2022-03-31

    Abstract: 本说明书实施例公开了一种模型剪枝方法、装置和计算机设备。所述方法包括:根据剪枝参数,确定掩码信息;所述掩码信息用于指示目标模型中剪枝对象的有效状态;将样本输入至增添了掩码信息后的目标模型,得到目标模型的第一输出;根据第一输出,优化参数信息;所述参数信息包括目标模型的模型参数和剪枝参数;迭代执行以上步骤,直至满足结束条件;根据掩码信息,对剪枝对象进行剪枝处理。本说明书实施例可以对目标模型进行剪枝处理,以减少存储资源和计算资源的占用。

Patent Agency Ranking