diff --git a/distiller/pruning/greedy_filter_pruning.py b/distiller/pruning/greedy_filter_pruning.py index 2b067d7ba3be1de57f3d233b393e642bcca14628..a40aaca0174439a5a929b9440a517f5ea000562c 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