[Unstructured_prune]Fix a bug in skip params function (#931)

* fix a bug in skip params func

paddleslim/dygraph/prune/unstructured_pruner.py

@@ -93,6 +93,8 @@ class UnstructuredPruner():
             if not self._should_prune_layer(sub_layer):
                 continue
             for param in sub_layer.parameters(include_sublayers=False):
+                if param.name in self.skip_params:
+                    continue
                 t_param = param.value().get_tensor()
                 v_param = np.array(t_param)
                 if self.local_sparsity:
@@ -111,8 +113,11 @@ class UnstructuredPruner():
 
     def _update_masks(self):
         for name, sub_layer in self.model.named_sublayers():
-            if not self._should_prune_layer(sub_layer): continue
+            if not self._should_prune_layer(sub_layer):
+                continue
             for param in sub_layer.parameters(include_sublayers=False):
+                if param.name in self.skip_params:
+                    continue
                 mask = self.masks.get(param.name)
                 if self.local_sparsity:
                     bool_tmp = (
@@ -243,7 +248,7 @@ class UnstructuredPruner():
             for param in sub_layer.parameters(include_sublayers=False):
                 cond = len(param.shape) == 4 and param.shape[
                     2] == 1 and param.shape[3] == 1
-                if not cond: skip_params.add(sub_layer.full_name())
+                if not cond: skip_params.add(param.name)
         return skip_params
 
     def _should_prune_layer(self, layer):

paddleslim/prune/unstructured_pruner.py

@@ -254,7 +254,6 @@ class UnstructuredPruner():
             if 'norm' in op.type() and 'grad' not in op.type():
                 for input in op.all_inputs():
                     skip_params.add(input.name())
-        print(skip_params)
         return skip_params
 
     def _get_skip_params_conv1x1(self, program):