From 4f984efe61ef3ca59b041f00bd76e53680245aed Mon Sep 17 00:00:00 2001 From: Double_V Date: Sun, 26 Sep 2021 17:34:51 +0800 Subject: [PATCH] add eps (#901) --- paddleslim/dygraph/prune/filter_pruner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddleslim/dygraph/prune/filter_pruner.py b/paddleslim/dygraph/prune/filter_pruner.py index b6bf4c9a..73c6f808 100644 --- a/paddleslim/dygraph/prune/filter_pruner.py +++ b/paddleslim/dygraph/prune/filter_pruner.py @@ -265,7 +265,7 @@ class FilterPruner(Pruner): baseline = eval_func() plan = self.prune_var(var_name, dims, ratio) pruned_metric = eval_func() - loss = (baseline - pruned_metric) / baseline + loss = (baseline - pruned_metric) / baseline + 1e-3 _logger.info("pruned param: {}; {}; loss={}".format( var_name, ratio, loss)) sensitivities[var_name][ratio] = loss -- GitLab