From e146524e668e4d7d6fdf798f80d33fa5d0b22f53 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: Sat, 30 Sep 2017 22:37:48 -0400
Subject: [PATCH] Update net.py

---
 lib/net.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/lib/net.py b/lib/net.py
index 096b255..2cde23d 100644
--- a/lib/net.py
+++ b/lib/net.py
@@ -96,6 +96,7 @@ def __init__(self, pt, model=None, phase = caffe.TEST, noTF=1, accname=None, gt_
             self.gt_net.copy_from(gt_model)
         self._points_dict_name = cfgs._points_dict_name
         if 0: self.show_acc('init')
+        """stores pruned values, which will be saved to caffemodel later (since Net couldn't be dynamically changed)"""
         self.WPQ={}
         self.nonWPQ = {}
         self.bottoms2ch = []
@@ -957,9 +958,7 @@ def finalmodel(self, WPQ=None, **kwargs):
         return self.linear(WPQ, **kwargs)
 
     def infer_pad_kernel(self, W, origin_name):
-        num_output = W.shape[0]
-        kernel_h = W.shape[2]
-        kernel_w = W.shape[3]
+        num_output, _, kernel_h, kernel_w = W.shape
         assert kernel_h in [3,1]
         assert kernel_w in [3,1]
         pad_h = 1 if kernel_h == 3 else 0