PrunneTorch类方法,配置剪枝过程中用户自定义的重要性评估函数。用户未自定义时,默认是L1正则作为重要性。
set_importance_evaluation_function(importance_evaluation_function)
参数名
输入/返回值
含义
使用限制
importance_evaluation_function
输入
自定义重要性评估函数,必须是可调用函数。
必选。
数据类型:函数。