analysis

功能说明

剪枝分析函数,只做剪枝过程中的分析操作,并返回剪枝后的参数量和剪枝信息。

函数原型

analysis(reserved_ratio, un_prune_list=[0,-1])

参数说明

参数名

输入/返回值

含义

使用限制

reserved_ratio

输入

剪枝参数量保留比例。

必选。

数据类型:Float。

取值范围0-1。

un_prune_list

输入

指定不剪枝的层,默认首尾不剪。

可选。

数据类型:list,元素必须是int或者string。

默认值为[0, -1]。

若元素是int,说明是第几层不剪(只计算剪枝的算子Conv2d和Linear)。

若是string,表明是算子在网络中的名字。