set_importance_evaluation_function

功能说明

PrunneTorch类方法,配置剪枝过程中用户自定义的重要性评估函数。用户未自定义时,默认是L1正则作为重要性。

函数原型

set_importance_evaluation_function(importance_evaluation_function)

参数说明

参数名

输入/返回值

含义

使用限制

importance_evaluation_function

输入

自定义重要性评估函数,必须是可调用函数。

必选。

数据类型:函数。