From da2ca46824a75117fc052598ea646d25a6d0f216 Mon Sep 17 00:00:00 2001 From: whs Date: Thu, 11 Jun 2020 17:28:43 +0800 Subject: [PATCH] Remove debug code in pruner (#342) --- paddleslim/prune/pruner.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/paddleslim/prune/pruner.py b/paddleslim/prune/pruner.py index 9bc2de09..7fbac719 100644 --- a/paddleslim/prune/pruner.py +++ b/paddleslim/prune/pruner.py @@ -123,15 +123,12 @@ class Pruner(): merge_pruned_params = {} for param, pruned_axis, pruned_idx in pruned_params: - print("{}\t{}\t{}".format(param, pruned_axis, len(pruned_idx))) if param not in merge_pruned_params: merge_pruned_params[param] = {} if pruned_axis not in merge_pruned_params[param]: merge_pruned_params[param][pruned_axis] = [] merge_pruned_params[param][pruned_axis].append(pruned_idx) - print("param name: stage.0.conv_layer.conv.weights; idx: {}".format( - merge_pruned_params["stage.0.conv_layer.conv.weights"][1])) for param_name in merge_pruned_params: for pruned_axis in merge_pruned_params[param_name]: pruned_idx = np.concatenate(merge_pruned_params[param_name][ -- GitLab