一种模型剪枝方法、装置、设备和存储介质

本发明公开了一种模型剪枝方法、装置、设备和存储介质，涉及模型剪枝技术领域。该方法包括：根据预先设定的剪枝比例和初始模型中各卷积层的重要度对初始模型进行剪枝，得到剪枝模型，其中，剪枝模型在测试数据集上的第一损失函数值满足第一阈值条件；对剪枝模型进行模型优化，得到优化模型，并在测试数据集上计算优化模型的第二损失函数值；在确定第二损失函数值不满足第二阈值条件的情况下，返回执行根据预先设定的剪枝比例和初始模型中各卷积层的重要度对初始模型进行剪枝，直至第二损失函数值满足第二阈值条件；将上一优化模型确定为目标模型。通过本发明实施例的技术方案可以在优化模型大小的同时提高剪枝后模型的精度。