训练用于图像识别的神经网络的方法和设备

    公开(公告)号:CN114358274A

    公开(公告)日:2022-04-15

    申请号:CN202110662202.2

    申请日:2021-06-15

    Abstract: 公开了一种训练用于图像识别的神经网络的方法和设备。所述方法包括:接收输入图像集;以输入图像集为输入执行预训练的神经网络,以获取推断的结果值,并基于与输入图像集对应的预定结果值和推断的结果值获取预训练的神经网络的第一任务准确度;基于通道单元,通过基于预设学习权重并基于与预训练的神经网络的多个层中的每个的通道对应的逐通道修剪参数调整多个通道的节点之间的权重,来修剪神经网络;基于第一任务准确度和修剪后的神经网络的任务准确度更新学习权重;基于更新后的学习权重和修剪后的神经网络的任务准确度更新逐通道修剪参数;和基于通道单元,基于更新后的学习权重并基于更新后的逐通道修剪参数重新修剪修剪后的神经网络。

Patent Agency Ranking