Skip to content
Snippets Groups Projects
Commit 516ba418 authored by Bob Wu's avatar Bob Wu Committed by Neta Zmora
Browse files

fix bug in greedy filter pruning (#286)

After del zeros_mask_dict is called, we use zeros_mask_dict again and error arose.
parent bf111e4e
No related branches found
No related tags found
No related merge requests found
...@@ -300,7 +300,6 @@ def greedy_pruner(pruned_model, app_args, fraction_to_prune, pruning_step, test_ ...@@ -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), name="greedy__{}__{:.1f}__{:.1f}".format(str(iteration).zfill(3), compute_density*100, prec1),
dir=msglogger.logdir) dir=msglogger.logdir)
del scheduler del scheduler
del zeros_mask_dict
msglogger.info("Iteration {}: top1-{:.2f} {} compute-{:.2f}".format(*results[0:4])) msglogger.info("Iteration {}: top1-{:.2f} {} compute-{:.2f}".format(*results[0:4]))
assert iteration > 0 assert iteration > 0
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment