剪枝函数，配置剪枝过程中的各项参数，并返回剪枝信息，可在评估过程根据剪枝信息进行剪枝。
prune(reserved_ratio, un_prune_list=[0,-1])
参数名 |
输入/返回值 |
含义 |
使用限制 |
|---|---|---|---|
reserved_ratio |
输入 |
剪枝参数量保留比例。 |
必选。 数据类型：Float。 取值范围0-1。 |
un_prune_list |
输入 |
指定不剪枝的层，默认首尾不剪。 |
可选。 数据类型：list，元素必须是int或者string。 默认值为[0, -1]。 若元素是int，说明是第几层不剪（只计算剪枝的算子Conv2d和Linear)。 若是string，表明是算子在网络中的名字。 |