Fix greedy pruner.

demo/sensitive_prune/greedy_prune.py

@@ -217,7 +217,7 @@ def compress(args):
             train(i, pruned_program)
             acc = test(i, pruned_val_program)
         print("iter:{}; pruned FLOPS: {}; acc: {}".format(
-            iter, float(base_flops - current_flops) / base_flops), acc)
+            iter, float(base_flops - current_flops) / base_flops, acc))
         pruner.save_checkpoint(pruned_program, pruned_val_program)
 
 

paddleslim/prune/sensitive.py

@@ -57,12 +57,15 @@ def sensitivity(program,
             if ratio in sensitivities[name]['pruned_percent']:
                 _logger.debug('{}, {} has computed.'.format(name, ratio))
                 ratio += step_size
+                pruned_times += 1
                 continue
             if baseline is None:
                 baseline = eval_func(graph.program)
 
             param_backup = {}
             pruner = Pruner()
+            _logger.info("sensitive - param: {}; ratios: {}".format(name,
+                                                                    ratio))
             pruned_program = pruner.prune(
                 program=graph.program,
                 scope=scope,