From ee1d160f75bd171d8f9b8afd17a65bcb0327a623 Mon Sep 17 00:00:00 2001
From: Bar <elhararb@gmail.com>
Date: Wed, 20 Feb 2019 22:34:15 +0200
Subject: [PATCH] Fix arguments in group_pruning_fn calls (#167)
This patch corrects overlapping between
magnitude_fn and block_shape arguments during
block pruning.
---
distiller/pruning/ranked_structures_pruner.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/distiller/pruning/ranked_structures_pruner.py b/distiller/pruning/ranked_structures_pruner.py
index d6d3fe1..0ae0c12 100755
--- a/distiller/pruning/ranked_structures_pruner.py
+++ b/distiller/pruning/ranked_structures_pruner.py
@@ -119,7 +119,8 @@ class LpRankedStructureParameterPruner(RankedStructureParameterPruner):
group_pruning_fn = partial(self.rank_and_prune_blocks, block_shape=self.block_shape)
binary_map = group_pruning_fn(fraction_to_prune, param, param_name,
- zeros_mask_dict, model, binary_map, self.magnitude_fn)
+ zeros_mask_dict, model, binary_map,
+ magnitude_fn=self.magnitude_fn)
return binary_map
@staticmethod
--
GitLab