From 516ba418864a0d8ddfd39da94f75281d4a8c9527 Mon Sep 17 00:00:00 2001
From: Bob Wu <bowen.wu@163.com>
Date: Fri, 14 Jun 2019 17:50:08 +0800
Subject: [PATCH] fix bug in greedy filter pruning (#286)

After del zeros_mask_dict is called, we use zeros_mask_dict again and error arose.
---
 distiller/pruning/greedy_filter_pruning.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/distiller/pruning/greedy_filter_pruning.py b/distiller/pruning/greedy_filter_pruning.py
index 2b067d7..a40aaca 100755
--- a/distiller/pruning/greedy_filter_pruning.py
+++ b/distiller/pruning/greedy_filter_pruning.py
@@ -300,7 +300,6 @@ def greedy_pruner(pruned_model, app_args, fraction_to_prune, pruning_step, test_
                         name="greedy__{}__{:.1f}__{:.1f}".format(str(iteration).zfill(3), compute_density*100, prec1),
                         dir=msglogger.logdir)
         del scheduler
-        del zeros_mask_dict
         msglogger.info("Iteration {}: top1-{:.2f} {} compute-{:.2f}".format(*results[0:4]))
 
     assert iteration > 0
-- 
GitLab