Skip to content

Commit ea4cd14

Browse files
author
Yihui He 何宜晖
authored
Update net.py
1 parent e146524 commit ea4cd14

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

lib/net.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1645,6 +1645,13 @@ def extractResB(a):
16451645
return residual_B
16461646

16471647
def dictionary_kernel(self, X_name, weights, d_prime, Y_name, Y):
1648+
""" channel pruning algorithm wrapper
1649+
X_name: the conv layer to prune
1650+
weights: deprecated
1651+
d_prime: number of perserving channels (c' in paper), the speed-up ratio = d_prime / number of channels
1652+
Y_name: the next conv layer (For later removing of corresponding pruned weights)
1653+
Y: deprecated
1654+
"""
16481655
# weights,Y is None
16491656
if not self._mem:
16501657
feats_dict, points_dict = self.extract_features([X_name, Y_name], save=1)

0 commit comments

Comments
 (0)