Fix pruner issues (#2265)

examples/model_compress/model_prune_torch.py

@@ -27,7 +27,7 @@ prune_config = {
         'model_name': 'naive',
         'pruner_class': AGP_Pruner,
         'config_list': [{
-            'initial_sparsity': 0,
+            'initial_sparsity': 0.,
             'final_sparsity': 0.8,
             'start_epoch': 0,
             'end_epoch': 10,

src/sdk/pynni/nni/compression/torch/pruners.py

@@ -212,7 +212,7 @@ class AGP_Pruner(Pruner):
         if epoch > 0:
             self.now_epoch = epoch
             for wrapper in self.get_modules_wrapper():
-                wrapper.if_calculated.copy_(torch.tensor(0)) # pylint: disable=not-callable
+                wrapper.if_calculated = False
 
 class SlimPruner(Pruner):
     """

test/scripts/model_compression.sh

@@ -19,6 +19,15 @@ do
     python3 model_prune_torch.py --pruner_name $name --pretrain_epochs 1 --prune_epochs 1
 done
 
+echo 'testing level pruner pruning'
+python3 model_prune_torch.py --pruner_name level --pretrain_epochs 1 --prune_epochs 1
+
+echo 'testing agp pruning'
+python3 model_prune_torch.py --pruner_name agp --pretrain_epochs 1 --prune_epochs 2
+
+echo 'testing mean_activation pruning'
+python3 model_prune_torch.py --pruner_name mean_activation --pretrain_epochs 1 --prune_epochs 1
+
 #echo "testing lottery ticket pruning..."
 #python3 lottery_torch_mnist_fc.py