set_node_reserved_ratio

功能说明

PrunneTorch类方法,配置模型剪枝过程中算子节点保留的参数比例。

函数原型

set_node_reserved_ratio(node_reserved_ratio=0.5)

参数说明

参数名

输入/返回值

含义

使用限制

node_reserved_ratio

输入

剪枝过程中最多裁剪掉的算子节点比例。

必选。

数据类型:Float。

取值范围0-1,默认值0.5。