From c69ae501cfb0f3085044915f58722b37d6ba1647 Mon Sep 17 00:00:00 2001 From: wanghaoshuang Date: Wed, 20 Nov 2019 11:18:30 +0800 Subject: [PATCH] Disable log of pruner. --- paddleslim/prune/pruner.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/paddleslim/prune/pruner.py b/paddleslim/prune/pruner.py index c7cc9c9e..0fdde525 100644 --- a/paddleslim/prune/pruner.py +++ b/paddleslim/prune/pruner.py @@ -113,7 +113,7 @@ class Pruner(): new_shape = list(ori_shape) new_shape[0] -= pruned_num param.set_shape(new_shape) - _logger.info("prune [{}] from {} to {}".format(param.name( + _logger.debug("prune [{}] from {} to {}".format(param.name( ), ori_shape, new_shape)) self.pruned_list[0].append(param.name()) return range(pruned_num) @@ -141,7 +141,7 @@ class Pruner(): new_shape = list(param.shape()) new_shape[0] = pruned_param.shape[0] param.set_shape(new_shape) - _logger.info("prune [{}] from {} to {}".format(param.name( + _logger.debug("prune [{}] from {} to {}".format(param.name( ), ori_shape, new_shape)) self.pruned_list[0].append(param.name()) return pruned_idx @@ -182,7 +182,7 @@ class Pruner(): new_shape = list(ori_shape) new_shape[pruned_axis] -= pruned_num param.set_shape(new_shape) - _logger.info("prune [{}] from {} to {}".format(param.name( + _logger.debug("prune [{}] from {} to {}".format(param.name( ), ori_shape, new_shape)) self.pruned_list[pruned_axis].append(param.name()) @@ -206,7 +206,7 @@ class Pruner(): new_shape = list(param.shape()) new_shape[pruned_axis] = pruned_param.shape[pruned_axis] param.set_shape(new_shape) - _logger.info("prune [{}] from {} to {}".format(param.name( + _logger.debug("prune [{}] from {} to {}".format(param.name( ), ori_shape, new_shape)) self.pruned_list[pruned_axis].append(param.name()) -- GitLab