Skip to content
Snippets Groups Projects
Unverified Commit 61fab0d0 authored by Neta Zmora's avatar Neta Zmora Committed by GitHub
Browse files

ranked_structures_pruner.py issue #363

When adding support for specifying the rounding function for pruning structures, the code of `ActivationRankedFilterPruner` was updated incorrectly (adding an extra argument).
This code is not covered by any of the tests and therefore the bug was not caught in time.  To do: add unit test.
parent 6e223f85
No related branches found
No related tags found
No related merge requests found
......@@ -446,8 +446,7 @@ class ActivationRankedFilterPruner(_RankedStructureParameterPruner):
if fraction_to_prune == 0:
return
binary_map = self.rank_and_prune_filters(fraction_to_prune, param, param_name,
zeros_mask_dict, model, binary_map,
self.rounding_fn)
zeros_mask_dict, model, binary_map)
return binary_map
def rank_and_prune_filters(self, fraction_to_prune, param, param_name, zeros_mask_dict, model, binary_map=None):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment