From fd5321b3f3be48a0c7713af82018afc1acebd275 Mon Sep 17 00:00:00 2001 From: Bai Yifan Date: Mon, 28 Oct 2019 09:51:36 +0800 Subject: [PATCH] modify slim print precision to round(,6), test=develop (#20833) --- python/paddle/fluid/contrib/slim/core/compressor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/contrib/slim/core/compressor.py b/python/paddle/fluid/contrib/slim/core/compressor.py index 41f282bf99..2ebcbc1fce 100644 --- a/python/paddle/fluid/contrib/slim/core/compressor.py +++ b/python/paddle/fluid/contrib/slim/core/compressor.py @@ -505,7 +505,7 @@ class Compressor(object): _logger.info("epoch:{}; batch_id:{}; {} = {}".format( context.epoch_id, context.batch_id, context.optimize_graph.out_nodes.keys( - ), [round(r, 3) for r in results])) + ), [round(r, 6) for r in results])) for strategy in self.strategies: strategy.on_batch_end(context) context.batch_id += 1 @@ -525,7 +525,7 @@ class Compressor(object): _logger.info("epoch:{}; batch_id:{}; {} = {}".format( context.epoch_id, context.batch_id, context.optimize_graph.out_nodes.keys( - ), [round(r, 3) for r in results])) + ), [round(r, 6) for r in results])) for strategy in self.strategies: strategy.on_batch_end(context) context.batch_id += 1 -- GitLab