Fix argument order to map

JJGO · Mar 9, 2020 · d755667 · d755667
1 parent 5ae677d
commit d755667
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/strategies/random.py b/strategies/random.py
@@ -20,5 +20,5 @@ class RandomPruning(VisionPruning):
 
     def model_masks(self):
         params = self.params()
-        masks = map_importances(params, lambda x: random_mask(x, self.fraction))
+        masks = map_importances(lambda x: random_mask(x, self.fraction), params)
         return masks