diff --git a/python/paddle/fluid/contrib/slim/core/compressor.py b/python/paddle/fluid/contrib/slim/core/compressor.py index 41f282bf997e0bc925920044c88a8e1e57b54c11..2ebcbc1fce36dd795ed1634238c632046b1ce4a2 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