diff --git a/paddleslim/prune/sensitive.py b/paddleslim/prune/sensitive.py index 1d5f6b00d5dbb8fd1e4546701ddba18dbdfb9227..3341a4a2e847bfb62b00a5caf4807df3b960ea68 100644 --- a/paddleslim/prune/sensitive.py +++ b/paddleslim/prune/sensitive.py @@ -75,10 +75,8 @@ def sensitivity(program, _logger.info("pruned param: {}; {}; loss={}".format(name, ratio, loss)) - for brother in pruner.pruned_list[0]: - if brother in sensitivities: - sensitivities[name]['pruned_percent'].append(ratio) - sensitivities[name]['loss'].append(loss) + sensitivities[name]['pruned_percent'].append(ratio) + sensitivities[name]['loss'].append(loss) _save_sensitivities(sensitivities, sensitivities_file) diff --git a/paddleslim/version.py b/paddleslim/version.py index f7ed0174edd9207b6d582008761420c12dec8018..3e95d57aa8f5b558d818dfff8b5d85daba2c6068 100644 --- a/paddleslim/version.py +++ b/paddleslim/version.py @@ -14,4 +14,4 @@ # limitations under the License. """ PaddleSlim version string """ __all__ = ["slim_version"] -slim_version = "4.0" +slim_version = "0.1"