diff --git a/paddleslim/prune/sensitive.py b/paddleslim/prune/sensitive.py index 8dd2f88b7de9eba62df447c78ef629a32111cd09..d0edbc2ba9f1e6709fea22136628836081e676a8 100644 --- a/paddleslim/prune/sensitive.py +++ b/paddleslim/prune/sensitive.py @@ -80,8 +80,12 @@ def sensitivity(program, loss = (baseline - pruned_metric) / baseline _logger.info("pruned param: {}; {}; loss={}".format(name, ratio, loss)) - sensitivities[name]['pruned_percent'].append(ratio) - sensitivities[name]['loss'].append(loss) + + for brother in pruner.pruned_list[0]: + if brother in sensitivities: + sensitivities[name]['pruned_percent'].append(ratio) + sensitivities[name]['loss'].append(loss) + _save_sensitivities(sensitivities, sensitivities_file) # restore pruned parameters