公开(公告)号：US20210312293A1

公开(公告)日：2021-10-07

申请号：US17354873

申请日：2021-06-22

Applicant: Beijing Baidu Netcom Science and Technology Co., Ltd.

Inventor： Teng Xi ,  Gang Zhang ,  Shengzhao Wen

IPC: G06N3/08 ,  G06N3/04

Abstract: Embodiments of the present disclosure provide a method and apparatus for determining a network model pruning strategy, a device and a storage medium. The method may include: generating a BN threshold search space using configuration information of the BN threshold search space for a network model for a target hardware; generating a pruning strategy code generator using the BN threshold search space; randomly generating a BN threshold code using the pruning strategy code generator; decoding the BN threshold code to obtain a candidate BN threshold; and determining a target pruning strategy of the network model for the target hardware, based on a pruning accuracy loss of the network model corresponding to the candidate BN threshold.