Update net.py

ethanhe42 · Oct 6, 2017 · ea4cd14 · ea4cd14
1 parent e146524
commit ea4cd14
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/lib/net.py b/lib/net.py
@@ -1645,6 +1645,13 @@ def extractResB(a):
         return residual_B
 
     def dictionary_kernel(self, X_name, weights, d_prime, Y_name, Y):
+        """ channel pruning algorithm wrapper
+        X_name: the conv layer to prune
+        weights: deprecated
+        d_prime: number of perserving channels (c' in paper), the speed-up ratio = d_prime / number of channels
+        Y_name: the next conv layer (For later removing of corresponding pruned weights)
+        Y: deprecated
+        """
         # weights,Y is None
         if not self._mem:
             feats_dict, points_dict = self.extract_features([X_name, Y_name], save=1)