【技术实现步骤摘要】
本申请属于网络模型,具体涉及一种网络模型裁剪方法及装置。
技术介绍
1、随着训练图像的数量越来越多,网络模型越来越深,模型的性能也越来越好。但是在网络性能提升的同时,网络的耗时也在逐步增加,网络运行所占用的磁盘空间和内存消耗也越来越多,因此训练得到的网络模型难以直接运用到嵌入式设备等移动端设备中去。对模型进行压缩与加速的技术得到广泛发展,而模型裁剪技术便是其中最广泛的研究方向之一。
2、模型结构化裁剪技术大多需要手动指定或通过某个指标来确定每个网络层的裁剪率,但是设定的裁剪率很难精确的满足嵌入式设备的耗时要求,得到的裁剪模型并非最优的裁剪结果。
技术实现思路
1、本申请实施例提供一种网络模型裁剪方法及装置,以解决现有的网络模型裁剪方式不能满足嵌入式设备的要求的问题。
2、第一方面,本申请实施例提供了一种网络模型裁剪方法,包括:
3、根据嵌入式平台对所运行待裁剪网络模型所要求的通道对齐方式,确定所述待裁剪网络模型至少一个卷积层内的待裁剪通道数量;
< ...【技术保护点】
1.一种网络模型裁剪方法,其特征在于,包括:
2.根据权利要求1所述的网络模型裁剪方法,其特征在于,所述根据嵌入式平台对所运行待裁剪网络模型所要求的通道对齐方式,确定所述待裁剪网络模型至少一个卷积层内的待裁剪通道数量,包括:
3.根据权利要求2所述的网络模型裁剪方法,其特征在于,所述计算每组通道所对应的裁剪概率,包括:
4.根据权利要求2所述的网络模型裁剪方法,其特征在于,所述根据每组通道所对应的裁剪概率确定所述卷积层内的待裁剪通道组数,包括:
5.根据权利要求3或4所述的网络模型裁剪方法,其特征在于,所述激活层为可微极
...【技术特征摘要】
1.一种网络模型裁剪方法,其特征在于,包括:
2.根据权利要求1所述的网络模型裁剪方法,其特征在于,所述根据嵌入式平台对所运行待裁剪网络模型所要求的通道对齐方式,确定所述待裁剪网络模型至少一个卷积层内的待裁剪通道数量,包括:
3.根据权利要求2所述的网络模型裁剪方法,其特征在于,所述计算每组通道所对应的裁剪概率,包括:
4.根据权利要求2所述的网络模型裁剪方法,其特征在于,所述根据每组通道所对应的裁剪概率确定所述卷积层内的待裁剪通道组数,包括:
5.根据权利要求3或4所述的网络模型裁剪方法,其特征在于,所述激活层为可微极化门,所述被裁剪的概率为0或1。
6.根据权利要求1所述的网...
【专利技术属性】
技术研发人员:张烁,颜聪泉,杨彭举,谢迪,
申请(专利权)人:杭州海康威视数字技术股份有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。