set_node_reserved_ratio
功能说明
PrunneTorch类方法,配置模型剪枝过程中算子节点保留的参数比例。
函数原型
set_node_reserved_ratio(node_reserved_ratio=0.5)
参数说明
参数名 |
输入/返回值 |
含义 |
使用限制 |
---|---|---|---|
node_reserved_ratio |
输入 |
剪枝过程中最多裁剪掉的算子节点比例。 |
必选。 数据类型:Float。 取值范围0-1,默认值0.5。 |
父主题: PruneTorch