From ea4cd1489dfa0a8f51c94402f054786cba9f6d38 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Yihui=20He=20=E4=BD=95=E5=AE=9C=E6=99=96?=
 <yihuihe@foxmail.com>
Date: Fri, 6 Oct 2017 03:49:49 -0400
Subject: [PATCH] Update net.py

---
 lib/net.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/lib/net.py b/lib/net.py
index 2cde23d..f76b77f 100644
--- 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)